| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | reduce_decomposer.cc | 36 Status HandleReduce(HloInstruction* hlo) override { in HandleReduce() function in xla::__anon95dee7e50111::VariadicReductionLayoutEqualizer
73 Status HandleReduce(HloInstruction* hlo) override { in HandleReduce() function in xla::__anon95dee7e50111::ReduceDecomposerVisitor
|
| H A D | tree_reduction_rewriter.cc | 45 Status HandleReduce(HloInstruction *hlo) override { in HandleReduce() function in xla::ReductionRewriterVisitor
|
| H A D | dfs_hlo_visitor_with_default.h | 196 Status HandleReduce(HloInstructionPtr reduce) override { in HandleReduce() function
|
| H A D | hlo_cost_analysis.cc | 385 Status HloCostAnalysis::HandleReduce(const HloInstruction* reduce) { in HandleReduce() function in xla::HloCostAnalysis
|
| H A D | hlo_verifier.cc | 1022 Status ShapeVerifier::HandleReduce(HloInstruction* reduce) { in HandleReduce() function in xla::ShapeVerifier
|
| H A D | dynamic_dimension_inference.cc | 444 Status DynamicDimensionInferenceVisitor::HandleReduce(HloInstruction* hlo) { in HandleReduce() function in xla::DynamicDimensionInferenceVisitor
|
| H A D | hlo_instruction_test.cc | 100 Status HandleReduce(HloInstruction* reduce) override { in HandleReduce() function in xla::__anon6a94cac00111::OpAndUserCollectingVisitor
|
| H A D | hlo_evaluator.cc | 3709 Status HloEvaluator::HandleReduce(HloInstruction* instr) { in HandleReduce() function in xla::HloEvaluator
|
| H A D | algebraic_simplifier.cc | 5279 Status AlgebraicSimplifierVisitor::HandleReduce(HloInstruction* hlo) { in HandleReduce() function in xla::AlgebraicSimplifierVisitor
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| H A D | reduction_dimension_grouper.cc | 36 Status HandleReduce(HloInstruction *hlo) override { in HandleReduce() function in xla::gpu::ReduceDimensionGroupVisitor
|
| H A D | reduction_splitter.cc | 30 Status HandleReduce(HloInstruction *reduce) override { in HandleReduce() function in xla::gpu::ReductionSplitterVisitor
|
| H A D | reduction_degenerate_dim_remover.cc | 39 Status HandleReduce(HloInstruction *hlo) override { in HandleReduce() function in xla::gpu::ReductionDegenerateDimRemoverVisitor
|
| H A D | reduction_layout_normalizer.cc | 38 Status HandleReduce(HloInstruction *hlo) override { in HandleReduce() function in xla::gpu::EnforceMinorToMajorReduceOpVisitor
|
| H A D | tree_reduction_rewriter.cc | 52 Status HandleReduce(HloInstruction *hlo) override { in HandleReduce() function in xla::gpu::ReductionRewriterVisitor
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | tile_ops.cc | 488 void HandleReduce(OpKernelContext* context, in HandleReduce() function in tensorflow::TileGradientOp
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| H A D | ir_emitter.cc | 1900 Status IrEmitter::HandleReduce(HloInstruction* reduce) { in HandleReduce() function in xla::cpu::IrEmitter
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| H A D | spmd_partitioner.cc | 3470 Status SpmdPartitioningVisitor::HandleReduce(HloInstruction* hlo) { in HandleReduce() function in xla::spmd::SpmdPartitioningVisitor
|