HomeSort by relevance Sort by last modified time
    Searched refs:reduce_op (Results 1 - 5 of 5) sorted by null

  /external/eigen/unsupported/Eigen/CXX11/src/Tensor/
TensorArgMax.h 174 const ReduceOp& reduce_op,
177 : m_xpr(expr), m_reduce_op(reduce_op), m_return_dim(return_dim), m_reduce_dims(reduce_dims) {}
184 const ReduceOp& reduce_op() const { return m_reduce_op; } function in class:Eigen::TensorTupleReducerOp
224 m_impl(op.expression().index_tuples().reduce(op.reduce_dims(), op.reduce_op()), device),
  /external/tensorflow/tensorflow/compiler/xla/tests/
reduce_test.cc 158 Computation reduce_op; local
161 reduce_op = CreateScalarAndComputation(&builder);
164 reduce_op = CreateScalarOrComputation(&builder);
167 builder.Reduce(input_pred, init_value, reduce_op,
    [all...]
  /external/tensorflow/tensorflow/contrib/estimator/python/estimator/
replicate_model_fn.py 722 reduce_op = state_ops.assign_add(metric_variables[i],
725 with ops_lib.control_dependencies([reduce_op]):
  /external/tensorflow/tensorflow/contrib/labeled_tensor/python/ops/
ops.py     [all...]
  /external/python/cpython3/Modules/
_pickle.c 3489 const char reduce_op = REDUCE; local
    [all...]

Completed in 422 milliseconds