Home
last modified time | relevance | path

Searched refs:call_target_name (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.h436 const string& call_target_name, absl::Span<const XlaOp> operands,
824 friend XlaOp CustomCall(XlaBuilder* builder, const string& call_target_name,
828 XlaBuilder* builder, const string& call_target_name,
1431 XlaOp CustomCall(XlaBuilder* builder, const string& call_target_name,
1440 XlaOp CustomCallWithLayout(XlaBuilder* builder, const string& call_target_name,
Dxla_builder.cc1494 const string& call_target_name, absl::Span<const XlaOp> operands, in CustomCall() argument
1499 if (absl::StartsWith(call_target_name, "$")) { in CustomCall()
1503 call_target_name); in CustomCall()
1506 instr.set_custom_call_target(call_target_name); in CustomCall()
3081 XlaOp CustomCall(XlaBuilder* builder, const string& call_target_name, in CustomCall() argument
3084 return builder->CustomCall(call_target_name, operands, shape, opaque, in CustomCall()
3088 XlaOp CustomCallWithLayout(XlaBuilder* builder, const string& call_target_name, in CustomCallWithLayout() argument
3092 return builder->CustomCall(call_target_name, operands, shape, opaque, in CustomCallWithLayout()
/external/tensorflow/tensorflow/compiler/xla/python/
Dlocal_computation_builder.cc619 const string& call_target_name, absl::Span<const LocalOp> operands, in CustomCall() argument
628 return xla::CustomCallWithLayout(&builder_, call_target_name, xla_ops, in CustomCall()
Dlocal_computation_builder.h311 LocalOp CustomCall(const string& call_target_name,
Dxla_client.py1419 call_target_name, argument
1439 return self._client.CustomCall(call_target_name, operands,