Home
last modified time | relevance | path

Searched defs:custom_call (Results 1 – 16 of 16) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcudnn_conv_rewriter.cc58 HloInstruction* custom_call = computation->AddInstruction( in CreateCudnnConv() local
501 HloInstruction* custom_call = [&]() -> HloInstruction* { in RunOnInstruction() local
Dcusolver_rewriter.cc122 HloInstruction* custom_call = in CreateCholesky() local
Dcudnn_conv_rewriter_test.cc309 const HloInstruction* custom_call = in TEST_F() local
444 const HloInstruction* custom_call = in TEST_F() local
Dir_emitter_unnested.cc373 Status IrEmitterUnnested::HandleCustomCall(HloInstruction* custom_call) { in HandleCustomCall()
/external/tensorflow/tensorflow/compiler/xla/service/interpreter/
Dcompiler.cc53 HloInstruction* custom_call, absl::Span<const Literal*> operands) { in HandleEvaluatorCustomCall()
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_evaluator.h148 std::function<StatusOr<Literal>(HloInstruction* custom_call, in set_custom_call_handler()
Dlayout_assignment_test.cc1208 const HloInstruction* custom_call = in TEST_F() local
1234 const HloInstruction* custom_call = in TEST_F() local
1268 const HloInstruction* custom_call = in TEST_F() local
1299 const HloInstruction* custom_call = FindInstruction(m.get(), "custom-call"); in TEST_F() local
Dlayout_assignment.cc421 const HloCustomCallInstruction* custom_call = in IsLayoutConstrainedCustomCall() local
468 const HloCustomCallInstruction* custom_call = in AddMandatoryConstraints() local
674 const HloCustomCallInstruction* custom_call = in CheckCustomCallLayout() local
Ddfs_hlo_visitor_with_default.h145 Status HandleCustomCall(HloInstructionPtr custom_call) override { in HandleCustomCall()
Dhlo_verifier.cc545 const HloCustomCallInstruction* custom_call = in HandleCustomCall() local
Dhlo_instruction.cc3443 if (auto custom_call = DynCast<HloCustomCallInstruction>(this)) { in convolution_dimension_numbers() local
3453 } else if (auto custom_call = DynCast<HloCustomCallInstruction>(this)) { in set_convolution_dimension_numbers() local
Dbuffer_assignment_test.cc1366 auto custom_call = builder.AddInstruction(HloInstruction::CreateCustomCall( in TEST_F() local
2360 HloInstruction* custom_call = main->GetInstructionWithName("custom_call"); in TEST_F() local
Dhlo_evaluator.cc1576 Status HloEvaluator::HandleCustomCall(HloInstruction* custom_call) { in HandleCustomCall()
Dhlo_evaluator_test.cc3211 [](HloInstruction* custom_call, absl::Span<const Literal*> operands) { in TEST_F()
3236 [](HloInstruction* custom_call, absl::Span<const Literal*> operands) { in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/tests/
Dcustom_call_test.cc175 auto custom_call = b.AddInstruction(HloInstruction::CreateCustomCall( in XLA_TEST_F() local
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emitter.cc2229 Status IrEmitter::HandleCustomCall(HloInstruction* custom_call) { in HandleCustomCall()