| /external/tensorflow/tensorflow/python/ops/ |
| D | math_ops.py | 1281 def _ReductionDims(x, axis, reduction_indices=None): # pylint: disable=invalid-name argument
1319 reduction_indices=None, argument
1466 reduction_indices=None, argument
1604 reduction_indices=None, argument
1857 reduction_indices=None, argument
1900 reduction_indices=None, argument
1979 reduction_indices=None, argument
2058 reduction_indices=None, argument
2155 reduction_indices=None, argument
2252 reduction_indices=None, argument
|
| D | string_ops.py | 308 def _reduce_join_reduction_dims(x, axis, reduction_indices): argument
333 reduction_indices=None, 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/cc/gradients/ |
| D | math_grad.cc | 802 auto reduction_indices = op.input(1); in MinOrMaxGrad() local
892 auto reduction_indices = Reshape(scope, op.input(1), {-1}); in ProdGrad() local
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | constant_folding.cc | 630 int reduction_indices = reduce_dims[j].size(); in MaterializeBroadcastGradientArgs() local
748 NodeDef* reduction_indices = graph_->add_node(); in MaterializeReductionIndices() local
2554 NodeDef* reduction_indices = node_map_->GetNode(reduction_indices_input); in SimplifyReduction() local
|
| D | layout_optimizer_test.cc | 785 auto reduction_indices = in TEST_F() local
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | common_shape_fns.cc | 1135 auto reduction_indices = reduction_indices_t->flat<T>(); in ReductionShapeHelper() local
|
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | propagate_fixed_sizes.cc | 574 const auto& reduction_indices = in ProcessTensorFlowReductionOperator() local
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | legacy_reference_ops.h | 863 const std::vector<int>& reduction_indices, T* output_data, in Mean()
|