| /external/tensorflow/tensorflow/python/ops/ |
| D | string_ops.py | 106 def _reduce_join_reduction_dims(x, axis, reduction_indices): argument
130 reduction_indices=None): argument
|
| D | math_ops.py | 1287 def _ReductionDims(x, axis, reduction_indices): argument
1310 def _may_reduce_to_scalar(keepdims, axis, reduction_indices, output): argument
1325 reduction_indices=None, argument
1387 reduction_indices=None, argument
1453 reduction_indices=None, argument
1525 reduction_indices=None, argument
1575 reduction_indices=None, argument
1624 reduction_indices=None, argument
1673 reduction_indices=None, argument
1731 reduction_indices=None, argument
[all …]
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | reduce_join_op_test.py | 103 reduction_indices, argument
129 reduction_indices, argument
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | reduce_join_op.cc | 76 gtl::InlinedVector<int32, 8> GetReducedIndices(const Tensor& reduction_indices, in GetReducedIndices()
129 const Tensor& reduction_indices = context->input(1); in Compute() local
|
| D | quantized_instance_norm.cc | 305 reduction_indices; in Compute() local
315 const Eigen::array<Index, 2> reduction_indices{1, 2}; in Compute() local
|
| /external/eigen/unsupported/test/ |
| D | cxx11_tensor_index_list.cpp | 319 ReductionIndices reduction_indices; in test_mixed_index_list() local
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | constant_folding.cc | 518 int reduction_indices = reduce_dims[j].size(); in MaterializeBroadcastGradientArgs() local
631 NodeDef* reduction_indices = graph_->add_node(); in MaterializeReductionIndices() local
|
| D | layout_optimizer_test.cc | 700 auto reduction_indices = in TEST_F() local
|
| /external/tensorflow/tensorflow/cc/gradients/ |
| D | math_grad.cc | 786 auto reduction_indices = op.input(1); in MinOrMaxGrad() local
876 auto reduction_indices = Reshape(scope, op.input(1), {-1}); in ProdGrad() local
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | common_shape_fns.cc | 1042 auto reduction_indices = reduction_indices_t->flat<T>(); in ReductionShapeHelper() local
|
| /external/tensorflow/tensorflow/contrib/lite/kernels/internal/reference/ |
| D | reference_ops.h | 2857 const std::vector<int>& reduction_indices, T* output_data, in Mean()
|
| /external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/ |
| D | optimized_ops.h | 4493 const std::vector<int>& reduction_indices, T* output_data, in Mean()
|