| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | memory_space_assignment_utils.cc | 93 if (auto* custom_call = in IsValueAllowedInAlternateMemory() local
|
| D | hlo_evaluator.h | 154 std::function<StatusOr<Literal>(HloInstruction* custom_call, in set_custom_call_handler()
|
| D | logical_buffer_analysis.cc | 192 Status LogicalBufferAnalysis::HandleCustomCall(HloInstruction* custom_call) { in HandleCustomCall()
|
| D | layout_assignment_test.cc | 1202 const HloInstruction* custom_call = in TEST_F() local
1228 const HloInstruction* custom_call = in TEST_F() local
1262 const HloInstruction* custom_call = in TEST_F() local
1293 const HloInstruction* custom_call = FindInstruction(m.get(), "custom-call"); in TEST_F() local
|
| D | layout_assignment.cc | 431 const HloCustomCallInstruction* custom_call = in IsLayoutConstrainedCustomCall() local
482 const HloCustomCallInstruction* custom_call = in AddMandatoryConstraints() local
691 const HloCustomCallInstruction* custom_call = in CheckCustomCallLayout() local
|
| D | dfs_hlo_visitor_with_default.h | 164 Status HandleCustomCall(HloInstructionPtr custom_call) override { in HandleCustomCall()
|
| D | tuple_points_to_analysis.cc | 478 Status TuplePointsToAnalysis::HandleCustomCall(HloInstruction* custom_call) { in HandleCustomCall()
|
| D | hlo_instruction.cc | 3860 if (auto* custom_call = DynCast<HloCustomCallInstruction>(this)) { in precision_config() local
4149 if (auto custom_call = DynCast<HloCustomCallInstruction>(this)) { in convolution_dimension_numbers() local
4159 } else if (auto custom_call = DynCast<HloCustomCallInstruction>(this)) { in set_convolution_dimension_numbers() local
|
| D | hlo_dataflow_analysis.cc | 436 HloInstruction* custom_call) { in UpdateCustomCallValueSet()
|
| D | hlo_cost_analysis.cc | 936 Status HloCostAnalysis::HandleCustomCall(const HloInstruction* custom_call) { in HandleCustomCall()
|
| D | hlo_verifier.cc | 793 const HloCustomCallInstruction* custom_call = in HandleCustomCall() local
|
| D | buffer_assignment_test.cc | 1538 auto custom_call = builder.AddInstruction(HloInstruction::CreateCustomCall( in TEST_F() local
2507 HloInstruction* custom_call = main->GetInstructionWithName("custom_call"); in TEST_F() local
|
| D | hlo_evaluator_test.cc | 4438 [](HloInstruction* custom_call, absl::Span<const Literal*> operands) { in TEST_F()
4463 [](HloInstruction* custom_call, absl::Span<const Literal*> operands) { in TEST_F()
|
| D | hlo_evaluator.cc | 2515 Status HloEvaluator::HandleCustomCall(HloInstruction* custom_call) { in HandleCustomCall()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | cusolver_rewriter.cc | 91 HloInstruction* custom_call = in CreateCholesky() local
|
| D | gpu_conv_rewriter_test.cc | 312 const HloInstruction* custom_call = in TEST_F() local
451 const HloInstruction* custom_call = in TEST_F() local
|
| D | gpu_conv_rewriter.cc | 59 HloInstruction* custom_call = computation->AddInstruction( in CreateGpuConv() local
|
| D | ir_emitter_unnested.cc | 1062 Status IrEmitterUnnested::HandleCustomCall(HloInstruction* custom_call) { in HandleCustomCall()
1535 auto custom_call = mlir::cast<mlir::lmhlo::CustomCallOp>(input.op); in EmitCustomCallThunkFromMlir() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/interpreter/ |
| D | compiler.cc | 54 HloInstruction* custom_call, absl::Span<const Literal*> operands) { in HandleEvaluatorCustomCall()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | mhlo_to_lhlo_with_xla.cc | 651 const HloCustomCallInstruction* custom_call) { in EmitCholesky()
661 const HloCustomCallInstruction* custom_call) { in EmitGemm()
725 const HloCustomCallInstruction* custom_call) { in EmitDnnConvolution()
846 const HloCustomCallInstruction* custom_call) { in EmitDnnBatchNorm()
|
| D | legalize_tf.cc | 5153 auto custom_call = rewriter.create<mhlo::CustomCallOp>( in matchAndRewrite() local
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | custom_call_test.cc | 175 auto custom_call = b.AddInstruction(HloInstruction::CreateCustomCall( in XLA_TEST_F() local
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ |
| D | hlo_function_importer.cc | 319 auto custom_call = Cast<HloCustomCallInstruction>(instruction); in ImportInstructionImpl() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | ir_emitter.cc | 2295 Status IrEmitter::HandleCustomCall(HloInstruction* custom_call) { in HandleCustomCall()
|
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | spmd_partitioner_test.cc | 2378 auto custom_call = FindInstruction(module.get(), "custom-call.1"); in TEST_F() local
|