Searched defs:call_target_name (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | ops.cc | 134 [](XlaBuilder* builder, const py::bytes& call_target_name, in BuildOpsSubmodule()
137 const py::bytes& opaque, bool has_side_effect) -> XlaOp { in BuildOpsSubmodule()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ir/ |
| D | mlir_hlo_builder.cc | 132 const string& call_target_name, absl::Span<const XlaOp> operands, in CustomCallInternal()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.cc | 1880 const string& call_target_name, absl::Span<const XlaOp> operands, in CustomCall()
1924 const string& call_target_name, absl::Span<const XlaOp> operands, in CustomCallInternal()
1959 const string& call_target_name, absl::Span<const XlaOp> operands, in CustomCall()
4464 XlaBuilder* builder, const string& call_target_name, in CustomCall()
4476 XlaBuilder* builder, const string& call_target_name, in CustomCallWithComputation()
4489 XlaBuilder* builder, const string& call_target_name, in CustomCallWithLayout()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter_unnested.cc | 1536 const std::string call_target_name = custom_call.call_target_name().str(); in EmitCustomCallThunkFromMlir() local
|