| /external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/ |
| D | reduce_test_util.cc | 50 Reduce operation = CreateReduce(axis, src.shape, OperationType::REDUCE_SUM, in ReduceSumChannelsIntTest()
91 CreateReduce(axis, src.shape, OperationType::REDUCE_PRODUCT, op_def, in ReduceProductChannelsUIntTest()
128 CreateReduce(axis, src_tensor.shape, OperationType::MEAN, op_def, in MeanHWTest()
155 CreateReduce(axis, src_tensor.shape, OperationType::REDUCE_SUM, in ReduceSumChannelsTest()
186 CreateReduce(axis, src_tensor.shape, OperationType::REDUCE_PRODUCT, in ReduceProductChannelsTest()
218 CreateReduce(axis, src_tensor.shape, OperationType::REDUCE_MAXIMUM, in ReduceMaxChannelsTest()
246 CreateReduce(axis, src_tensor.shape, OperationType::REDUCE_MINIMUM, in ReduceMinChannelsTest()
|
| D | reduce.h | 60 Reduce CreateReduce(const std::set<Axis>& axis_to_reduce, const BHWC& src_shape,
64 Reduce CreateReduce(const std::set<Axis>& axis_to_reduce,
|
| D | reduce.cc | 525 Reduce CreateReduce(const std::set<Axis>& axis_to_reduce, const BHWC& src_shape, in CreateReduce() function
532 Reduce CreateReduce(const std::set<Axis>& axis_to_reduce, in CreateReduce() function
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | tree_reduction_rewriter.cc | 209 HloInstruction::CreateReduce( in RewriteReduction()
226 std::unique_ptr<HloInstruction> outer_reduce = HloInstruction::CreateReduce( in RewriteReduction()
251 hlo->parent()->AddInstruction(HloInstruction::CreateReduce( in RewriteBatchDimensionLargerThanTile()
256 std::unique_ptr<HloInstruction> out = HloInstruction::CreateReduce( in RewriteBatchDimensionLargerThanTile()
|
| D | reduction_splitter.cc | 86 std::unique_ptr<HloInstruction> pre_reduce = HloInstruction::CreateReduce( in HandleReduce()
102 std::unique_ptr<HloInstruction> final_reduce = HloInstruction::CreateReduce( in HandleReduce()
|
| D | reduction_dimension_grouper.cc | 96 std::unique_ptr<HloInstruction> new_reduce = HloInstruction::CreateReduce( in HandleReduce()
|
| D | reduction_degenerate_dim_remover.cc | 90 std::unique_ptr<HloInstruction> new_reduce = HloInstruction::CreateReduce( in HandleReduce()
|
| D | horizontal_input_fusion_test.cc | 126 auto reduce = builder.AddInstruction(HloInstruction::CreateReduce( in TEST_F()
|
| D | reduction_layout_normalizer.cc | 150 std::unique_ptr<HloInstruction> new_reduce = HloInstruction::CreateReduce( in HandleReduce()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | batchnorm_expander.cc | 215 auto sum = add(HloInstruction::CreateReduce(feature_shape, operand, zero, in HandleBatchNormTraining()
220 auto squared_sum = add(HloInstruction::CreateReduce( in HandleBatchNormTraining()
483 add(HloInstruction::CreateReduce( in HandleBatchNormGrad()
488 auto grad_beta = add(HloInstruction::CreateReduce( in HandleBatchNormGrad()
|
| D | tree_reduction_rewriter.cc | 103 HloInstruction::CreateReduce(reduce_shape, reduce_window, initial_value, in HandleReduce()
|
| D | hlo_dce_test.cc | 378 builder.AddInstruction(HloInstruction::CreateReduce( in TEST_F()
419 builder.AddInstruction(HloInstruction::CreateReduce( in TEST_F()
429 builder.AddInstruction(HloInstruction::CreateReduce( in TEST_F()
|
| D | dynamic_dimension_inference_test.cc | 224 auto reduce = builder.AddInstruction(HloInstruction::CreateReduce( in TEST_F()
257 HloInstruction::CreateReduce(reduce_shape, negate, init, {1}, GetAdd())); in TEST_F()
299 auto reduce = builder.AddInstruction(HloInstruction::CreateReduce( in TEST_F()
|
| D | hlo_creation_utils.cc | 393 HloInstruction::CreateReduce(result_shape, operand, init_value, in MakeReduceHlo()
471 HloInstruction::CreateReduce(output_shape, operands, init_values, in MakeReduceHlo()
|
| D | hlo_instruction.h | 913 static std::unique_ptr<HloInstruction> CreateReduce(
928 static std::unique_ptr<HloInstruction> CreateReduce(
937 static std::unique_ptr<HloInstruction> CreateReduce(
|
| D | bfloat16_normalization_test.cc | 206 HloInstruction* reduce = builder.AddInstruction(HloInstruction::CreateReduce( in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/ |
| D | gpu_ldg_test.cc | 132 HloInstruction* reduce = builder.AddInstruction(HloInstruction::CreateReduce( in TEST_F()
|
| D | parallel_reduction_test.cc | 148 fusion_builder.AddInstruction(HloInstruction::CreateReduce( in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/tests/ |
| D | cpu_fusion_test.cc | 168 auto reduce = builder.AddInstruction(HloInstruction::CreateReduce( in TEST_F()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/selectors/ |
| D | simple_selectors.cc | 199 CreateReduce(axis_to_reduce, src_shape, op_type, op_def, gpu_info)); in SelectReduce()
|
| /external/armnn/src/backends/neon/ |
| D | NeonWorkloadFactory.hpp | 301 std::unique_ptr<IWorkload> CreateReduce(const ReduceQueueDescriptor& descriptor,
|
| /external/armnn/include/armnn/backends/ |
| D | WorkloadFactory.hpp | 335 virtual std::unique_ptr<IWorkload> CreateReduce(const ReduceQueueDescriptor& descriptor,
|
| /external/armnn/src/backends/cl/ |
| D | ClWorkloadFactory.hpp | 304 std::unique_ptr<IWorkload> CreateReduce(const ReduceQueueDescriptor& descriptor,
|
| /external/armnn/src/backends/reference/ |
| D | RefWorkloadFactory.hpp | 320 std::unique_ptr<IWorkload> CreateReduce(const ReduceQueueDescriptor& descriptor,
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | cpu_gpu_fusion_test.cc | 609 auto reduce2 = builder.AddInstruction(HloInstruction::CreateReduce( in XLA_TEST_F()
629 auto reduce2 = builder.AddInstruction(HloInstruction::CreateReduce( in XLA_TEST_F()
|