Searched defs:custom_call_target (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_instruction.cc | 1880 absl::string_view custom_call_target, std::string opaque, in CreateCustomCall()
1888 HloComputation* to_apply, absl::string_view custom_call_target, in CreateCustomCall()
1898 absl::string_view custom_call_target, std::string opaque, in CreateCustomCall()
1907 absl::string_view custom_call_target, in CreateCustomCall()
4785 const std::string& HloInstruction::custom_call_target() const { in custom_call_target() function in xla::HloInstruction
|
| D | hlo.proto | 194 string custom_call_target = 28; field
|
| D | hlo_instructions.cc | 2641 absl::string_view custom_call_target, std::string opaque, in HloCustomCallInstruction()
2657 HloComputation* to_apply, absl::string_view custom_call_target, in HloCustomCallInstruction()
2675 absl::string_view custom_call_target, std::string opaque, in HloCustomCallInstruction()
2695 absl::string_view custom_call_target, std::string opaque, in HloCustomCallInstruction()
|
| D | hlo_instructions.h | 1845 const std::string& custom_call_target() const { return custom_call_target_; } in custom_call_target() function
|
| D | hlo_parser.cc | 2583 optional<std::string> custom_call_target; in CreateInstruction() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter_unnested.cc | 1334 CustomCallThunk::CustomCallTarget custom_call_target; in EmitCustomCallThunk() local
|