| /external/tensorflow/tensorflow/core/kernels/ |
| D | reduce_join_op.cc | 77 gtl::InlinedVector<int32, 8> GetReducedIndices(const Tensor& reduction_indices, in GetReducedIndices()
130 const Tensor& reduction_indices = context->input(1); in Compute() local
|
| D | quantized_instance_norm.cc | 312 reduction_indices; in Compute() local
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | math_ops.py | 2178 reduction_indices=None, argument
2385 reduction_indices=None, argument
2524 reduction_indices=None, argument
2817 reduction_indices=None, argument
2876 reduction_indices=None, argument
3004 reduction_indices=None, argument
3129 reduction_indices=None, argument
3235 reduction_indices=None, argument
3341 reduction_indices=None, argument
|
| D | string_ops.py | 312 reduction_indices=None, argument
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_ops_tensor_helper.cc | 62 Type InferReductionOpType(Value input, Value reduction_indices, in InferReductionOpType()
|
| D | tf_ops_a_m.cc | 3114 Value reduction_indices, BoolAttr keep_dims) { in build()
|
| D | tf_ops_n_z.cc | 1681 Value reduction_indices, BoolAttr keep_dims) { in build()
|
| /external/tensorflow/tensorflow/compiler/mlir/python/mlir_wrapper/ |
| D | ops.cc | 98 bool keep_dims = false) -> mlir::Operation* { in init_ops()
|
| /external/eigen/unsupported/test/ |
| D | cxx11_tensor_index_list.cpp | 318 ReductionIndices reduction_indices; in test_mixed_index_list() local
|
| /external/tensorflow/tensorflow/c/experimental/ops/ |
| D | math_ops.cc | 148 AbstractTensorHandle* const reduction_indices, in Sum()
|
| /external/tensorflow/tensorflow/dtensor/mlir/expansions/ |
| D | softmax_spmd_expander.cc | 47 const mlir::Value reduction_indices = in ComputeGlobalReduce() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | legalize_hlo.cc | 1831 ConstOp reduction_indices, in rewriteNonMatchInitValue()
1846 mhlo::ReduceOp reduce_op, Value input, ConstOp reduction_indices, in rewriteNonMatchInitValue()
1853 mhlo::ReduceOp reduce_op, Value input, ConstOp reduction_indices, in rewriteNonMatchInitValue()
1891 auto reduction_indices = rewriter.create<ConstOp>( in matchAndRewrite() local
2074 auto reduction_indices = rewriter.create<ConstOp>( in matchAndRewrite() local
|
| /external/tensorflow/tensorflow/cc/gradients/ |
| D | math_grad.cc | 790 auto reduction_indices = op.input(1); in MinOrMaxGrad() local
880 auto reduction_indices = Reshape(scope, op.input(1), {-1}); in ProdGrad() local
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | constant_folding.cc | 688 int reduction_indices = reduce_dims[j].size(); in MaterializeBroadcastGradientArgs() local
807 NodeDef* reduction_indices = graph_->add_node(); in MaterializeReductionIndices() local
2899 NodeDef* reduction_indices = node_map_->GetNode(reduction_indices_input); in SimplifyReduction() local
|
| /external/tensorflow/tensorflow/core/transforms/constant_folding/ |
| D | pass.cc | 1061 int reduction_indices = reduce_dims[j].size(); in matchAndRewrite() local
1973 Operation *reduction_indices = op->getOperand(1).getDefiningOp(); in matchAndRewrite() local
|
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | propagate_fixed_sizes.cc | 575 const auto& reduction_indices = in ProcessTensorFlowReductionOperator() local
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | common_shape_fns.cc | 1900 auto reduction_indices = reduction_indices_t->flat<T>(); in ReductionShapeHelper() local
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | legacy_reference_ops.h | 957 const std::vector<int>& reduction_indices, T* output_data, in Mean()
|