Home
last modified time | relevance | path

Searched refs:DefaultAction (Results 1 – 25 of 34) sorted by relevance

12

/external/tensorflow/tensorflow/compiler/xla/service/
Ddfs_hlo_visitor_with_default.h52 virtual Status DefaultAction(HloInstructionPtr hlo_instruction) = 0;
55 return DefaultAction(hlo); in HandleElementwiseUnary()
58 return DefaultAction(hlo); in HandleElementwiseBinary()
62 return DefaultAction(hlo); in HandleBatchNormTraining()
66 return DefaultAction(hlo); in HandleBatchNormInference()
70 return DefaultAction(hlo); in HandleBatchNormGrad()
74 return DefaultAction(clamp); in HandleClamp()
77 return DefaultAction(concatenate); in HandleConcatenate()
80 return DefaultAction(select); in HandleSelect()
83 return DefaultAction(tuple_select); in HandleTupleSelect()
[all …]
Dbfloat16_conversion_folding.cc35 Status DefaultAction(HloInstruction* hlo) override;
152 Status BFloat16ConversionFoldingVisitor::DefaultAction(HloInstruction* hlo) { in DefaultAction() function in xla::BFloat16ConversionFoldingVisitor
186 TF_RETURN_IF_ERROR(DefaultAction(crs)); in HandleAllReduce()
Ddfs_hlo_visitor_with_default_test.cc40 Status DefaultAction(HloInstruction* hlo) override { in TEST_F() function in xla::__anon1ba6c03c0111::TEST_F::ElementwiseTestVisitor
Dmap_inliner.cc42 Status DefaultAction(HloInstruction* /*hlo_instruction*/) override { in DefaultAction() function in xla::MapInlinerVisitor
Dlogical_buffer_analysis.h58 Status DefaultAction(HloInstruction* hlo_instruction) override;
Dbfloat16_normalization.cc36 Status DefaultAction(HloInstruction* hlo) override;
374 Status BFloat16NormalizationVisitor::DefaultAction(HloInstruction* hlo) { in DefaultAction() function in xla::BFloat16NormalizationVisitor
Dcall_inliner.cc43 Status DefaultAction(HloInstruction* hlo) override { in DefaultAction() function in xla::__anona1e1e8bc0111::SubcomputationInsertionVisitor
Dlogical_buffer_analysis.cc100 Status LogicalBufferAnalysis::DefaultAction(HloInstruction* hlo_instruction) { in DefaultAction() function in xla::LogicalBufferAnalysis
Ddynamic_dimension_inference.cc33 Status DefaultAction(HloInstruction* hlo) override;
98 Status DynamicDimensionInferenceVisitor::DefaultAction(HloInstruction* hlo) { in DefaultAction() function in xla::DynamicDimensionInferenceVisitor
Dtuple_points_to_analysis.h246 Status DefaultAction(HloInstruction* hlo_instruction) override;
Dhlo_evaluator.h176 Status DefaultAction(HloInstruction* hlo) override { in DefaultAction() function
Dconvolution_group_converter.cc47 Status DefaultAction(HloInstruction* /*hlo_instruction*/) override { in DefaultAction() function in xla::__anon9c1c05530111::ConvolutionVisitor
Dhlo_evaluator.cc1303 return DefaultAction(select); in HandleSelect()
1562 return DefaultAction(reduce); in HandleReduce()
1579 return DefaultAction(custom_call); in HandleCustomCall()
/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
Dfused_ir_emitter.h71 Status DefaultAction(HloInstruction* hlo) override;
Dfused_ir_emitter.cc43 Status FusedIrEmitter::DefaultAction(HloInstruction* hlo) { in DefaultAction() function in xla::FusedIrEmitter
/external/tensorflow/tensorflow/compiler/xla/tools/
Ddumped_computation_to_operation_list.cc45 Status DefaultAction(HloInstruction* hlo) override { in DefaultAction() function in xla::tools::OperationDumper
Dhlo_extractor.cc59 Status DefaultAction(const HloInstruction* hlo) override { in DefaultAction() function in xla::__anonc6eaaf4e0111::ExtractionVisitor
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emitter.cc228 return DefaultAction(copy); in HandleCopy()
308 return DefaultAction(select); in HandleSelect()
742 return DefaultAction(reduce_window); in HandleReduceWindow()
1253 return DefaultAction(convolution); in HandleConvolution()
1897 return DefaultAction(reduce); in HandleReduce()
1924 return DefaultAction(slice); in HandleSlice()
1929 return DefaultAction(slice); in HandleSlice()
1964 return DefaultAction(slice); in HandleSlice()
2060 return DefaultAction(dynamic_slice); in HandleDynamicSlice()
2077 return DefaultAction(dynamic_update_slice); in HandleDynamicUpdateSlice()
[all …]
Dcpu_compiler.cc206 Status DefaultAction(HloInstruction* hlo_instruction) override { in DefaultAction() function in xla::cpu::__anonaf93c4290211::CollectProfileCandidates
213 TF_RETURN_IF_ERROR(DefaultAction(call)); in HandleCall()
227 TF_RETURN_IF_ERROR(DefaultAction(xla_while)); in HandleWhile()
Dir_emitter.h145 Status DefaultAction(HloInstruction* hlo) override;
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter.h77 Status DefaultAction(HloInstruction* hlo) override;
Dcudnn_batchnorm_rewriter.cc35 Status DefaultAction(HloInstruction* /*hlo_instruction*/) override { in DefaultAction() function in xla::gpu::__anondf8c4db30111::Visitor
Dir_emitter.cc68 Status IrEmitter::DefaultAction(HloInstruction* hlo) { in DefaultAction() function in xla::gpu::IrEmitter
424 return IrEmitter::DefaultAction(select); in HandleSelect()
Dir_emitter_unnested.h157 Status DefaultAction(HloInstruction* hlo) override;
/external/v8/src/torque/
Dearley-parser.h135 inline base::Optional<ParseResult> DefaultAction( in DefaultAction() function
146 Action action = DefaultAction)

12