| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | dfs_hlo_visitor_with_default.h | 53 virtual Status DefaultAction(HloInstructionPtr hlo_instruction) = 0;
56 return DefaultAction(hlo); in HandleElementwiseUnary()
59 return DefaultAction(hlo); in HandleElementwiseBinary()
63 return DefaultAction(hlo); in HandleBatchNormTraining()
67 return DefaultAction(hlo); in HandleBatchNormInference()
71 return DefaultAction(hlo); in HandleBatchNormGrad()
75 return DefaultAction(clamp); in HandleClamp()
78 return DefaultAction(concatenate); in HandleConcatenate()
81 return DefaultAction(select); in HandleSelect()
84 return DefaultAction(tuple_select); in HandleTupleSelect()
[all …]
|
| D | bfloat16_conversion_folding.cc | 40 Status DefaultAction(HloInstruction* hlo) override;
161 Status BFloat16ConversionFoldingVisitor::DefaultAction(HloInstruction* hlo) { in DefaultAction() function in xla::BFloat16ConversionFoldingVisitor
197 TF_RETURN_IF_ERROR(DefaultAction(crs)); in HandleAllReduce()
|
| D | dfs_hlo_visitor_with_default_test.cc | 40 Status DefaultAction(HloInstruction* hlo) override { in TEST_F() function in xla::__anonda2535f30111::TEST_F::ElementwiseTestVisitor
|
| D | map_inliner.cc | 42 Status DefaultAction(HloInstruction* /*hlo_instruction*/) override { in DefaultAction() function in xla::MapInlinerVisitor
|
| D | logical_buffer_analysis.h | 59 Status DefaultAction(HloInstruction* hlo_instruction) override;
|
| D | call_inliner.cc | 43 Status DefaultAction(HloInstruction* hlo) override { in DefaultAction() function in xla::__anon2bad62130111::SubcomputationInsertionVisitor
|
| D | logical_buffer_analysis.cc | 102 Status LogicalBufferAnalysis::DefaultAction(HloInstruction* hlo_instruction) { in DefaultAction() function in xla::LogicalBufferAnalysis
|
| D | bfloat16_normalization.cc | 46 Status DefaultAction(HloInstruction* hlo) override;
489 Status BFloat16NormalizationVisitor::DefaultAction(HloInstruction* hlo) { in DefaultAction() function in xla::__anon96c306cc0111::BFloat16NormalizationVisitor
|
| D | tuple_points_to_analysis.h | 247 Status DefaultAction(HloInstruction* hlo_instruction) override;
|
| D | hlo_evaluator.h | 190 Status DefaultAction(HloInstruction* hlo) override { in DefaultAction() function
|
| D | dynamic_padder.cc | 1566 Status DefaultAction(HloInstruction* hlo) override;
1685 Status DynamicShapeRemovingVisitor::DefaultAction(HloInstruction* hlo) { in DefaultAction() function in xla::__anon6d34db3b0111::DynamicShapeRemovingVisitor
1767 return DefaultAction(hlo); in HandleCustomCall()
|
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | spmd_partitioner.cc | 1368 Status SpmdPartitioningVisitor::DefaultAction(HloInstruction* hlo) { in DefaultAction() function in xla::spmd::SpmdPartitioningVisitor
1488 return DefaultAction(hlo); in HandleConcatenate()
1582 return DefaultAction(hlo); in HandleSlice()
1606 return DefaultAction(hlo); in HandleSlice()
1746 return DefaultAction(hlo); in HandleSort()
1751 return DefaultAction(hlo); in HandleSort()
1756 return DefaultAction(hlo); in HandleSort()
1760 return DefaultAction(hlo); in HandleSort()
1803 return DefaultAction(hlo); in HandleCustomCall()
1807 return DefaultAction(hlo); in HandleCustomCall()
[all …]
|
| D | fft_handler.cc | 353 return DefaultAction(hlo); in HandleFft()
360 return DefaultAction(hlo); in HandleFft()
367 return DefaultAction(hlo); in HandleFft()
|
| D | gather_scatter_handler.cc | 538 return DefaultAction(hlo); in HandleScatter()
561 return DefaultAction(hlo); in HandleScatter()
668 return DefaultAction(hlo); in HandleScatter()
702 return DefaultAction(gather); in HandleGather()
|
| D | spmd_partitioner.h | 408 Status DefaultAction(HloInstruction* hlo) override;
|
| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| D | fused_ir_emitter.cc | 43 Status FusedIrEmitter::DefaultAction(const HloInstruction* hlo) { in DefaultAction() function in xla::FusedIrEmitter
187 TF_RETURN_IF_ERROR(DefaultAction(instr)); in GetGenerator()
|
| D | fused_ir_emitter.h | 79 Status DefaultAction(const HloInstruction* hlo);
|
| /external/tensorflow/tensorflow/compiler/xla/tools/ |
| D | dumped_computation_to_operation_list.cc | 45 Status DefaultAction(HloInstruction* hlo) override { in DefaultAction() function in xla::tools::OperationDumper
|
| D | hlo_extractor.cc | 60 Status DefaultAction(const HloInstruction* hlo) override { in DefaultAction() function in xla::__anon2ff0d9c50111::ExtractionVisitor
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter.h | 78 Status DefaultAction(HloInstruction* hlo) override;
|
| D | cudnn_batchnorm_rewriter.cc | 37 Status DefaultAction(HloInstruction* /*hlo_instruction*/) override { in DefaultAction() function in xla::gpu::__anon8335950a0111::Visitor
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | ir_emitter.cc | 292 return DefaultAction(copy); in HandleCopy()
372 return DefaultAction(select); in HandleSelect()
661 return DefaultAction(reduce_window); in HandleReduceWindow()
1018 return DefaultAction(convolution); in HandleConvolution()
1815 return DefaultAction(reduce); in HandleReduce()
1838 return DefaultAction(slice); in HandleSlice()
1843 return DefaultAction(slice); in HandleSlice()
1878 return DefaultAction(slice); in HandleSlice()
1974 return DefaultAction(dynamic_slice); in HandleDynamicSlice()
1991 return DefaultAction(dynamic_update_slice); in HandleDynamicUpdateSlice()
[all …]
|
| D | cpu_compiler.cc | 242 Status DefaultAction(HloInstruction* hlo_instruction) override { in DefaultAction() function in xla::cpu::__anonba9a11000311::CollectProfileCandidates
249 TF_RETURN_IF_ERROR(DefaultAction(call)); in HandleCall()
263 TF_RETURN_IF_ERROR(DefaultAction(xla_while)); in HandleWhile()
|
| D | ir_emitter.h | 130 Status DefaultAction(HloInstruction* hlo) override;
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | mhlo_to_lhlo_with_xla.h | 181 tensorflow::Status DefaultAction(const xla::HloInstruction* instr) final;
|