Searched defs:call_target_name (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | ops.cc | 174 [](XlaBuilder* builder, const py::bytes& call_target_name, in BuildOpsSubmodule()
193 [](XlaBuilder* builder, const py::bytes& call_target_name, in BuildOpsSubmodule()
200 CustomCallApiVersion api_version) -> XlaOp { in BuildOpsSubmodule()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.cc | 1966 const std::string& call_target_name, absl::Span<const XlaOp> operands, in CustomCall()
2013 const std::string& call_target_name, absl::Span<const XlaOp> operands, in CustomCallInternal()
2074 const std::string& call_target_name, absl::Span<const XlaOp> operands, in CustomCall()
4421 XlaBuilder* builder, const std::string& call_target_name, in CustomCall()
4436 XlaBuilder* builder, const std::string& call_target_name, in CustomCallWithComputation()
4450 XlaBuilder* builder, const std::string& call_target_name, in CustomCallWithLayout()
4465 XlaBuilder* builder, const std::string& call_target_name, in CustomCallWithConvDnums()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ir/ |
| D | mlir_hlo_builder.cc | 143 const std::string& call_target_name, absl::Span<const XlaOp> operands, in CustomCallInternal()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | jitrt_custom_calls.cc | 1507 StringRef call_target_name, int32_t api_version, in operator ()()
|
| D | ir_emitter_unnested.cc | 1279 const std::string call_target_name = custom_call.getCallTargetName().str(); in EmitCustomCallThunk() local
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 5985 NamedAttribute call_target_name = rewriter.getNamedAttr( in matchAndRewrite() local
|