Searched defs:call_target_name (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | ops.cc | 155 [](XlaBuilder* builder, const py::bytes& call_target_name, in BuildOpsSubmodule()
174 [](XlaBuilder* builder, const py::bytes& call_target_name, in BuildOpsSubmodule()
181 CustomCallApiVersion api_version) -> XlaOp { in BuildOpsSubmodule()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.cc | 1892 const string& call_target_name, absl::Span<const XlaOp> operands, in CustomCall()
1939 const string& call_target_name, absl::Span<const XlaOp> operands, in CustomCallInternal()
1984 const string& call_target_name, absl::Span<const XlaOp> operands, in CustomCall()
4276 XlaBuilder* builder, const string& call_target_name, in CustomCall()
4291 XlaBuilder* builder, const string& call_target_name, in CustomCallWithComputation()
4305 XlaBuilder* builder, const string& call_target_name, in CustomCallWithLayout()
4320 XlaBuilder* builder, const string& call_target_name, in CustomCallWithConvDnums()
|
| /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/service/gpu/ |
| D | ir_emitter_unnested.cc | 1422 const std::string call_target_name = custom_call.call_target_name().str(); in EmitCustomCallThunk() local
|