Searched defs:custom_call_target (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_instruction.cc | 1757 absl::string_view custom_call_target, string opaque, in CreateCustomCall()
1765 HloComputation* to_apply, absl::string_view custom_call_target, in CreateCustomCall()
1775 absl::string_view custom_call_target, string opaque, in CreateCustomCall()
1784 absl::string_view custom_call_target, in CreateCustomCall()
4510 const string& HloInstruction::custom_call_target() const { in custom_call_target() function in xla::HloInstruction
|
| D | hlo.proto | 174 string custom_call_target = 28; field
|
| D | hlo_instructions.cc | 2478 absl::string_view custom_call_target, string opaque, in HloCustomCallInstruction()
2497 HloComputation* to_apply, absl::string_view custom_call_target, in HloCustomCallInstruction()
2519 absl::string_view custom_call_target, string opaque, in HloCustomCallInstruction()
2541 absl::string_view custom_call_target, string opaque, in HloCustomCallInstruction()
|
| D | hlo_instructions.h | 1530 const string& custom_call_target() const { return custom_call_target_; } in custom_call_target() function
|
| D | hlo_parser.cc | 2374 optional<std::string> custom_call_target; in ParseInstructionRhs() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter_unnested.cc | 1479 CustomCallThunk::CustomCallTarget custom_call_target; in EmitCustomCallThunk() local
|