Searched refs:SendWithToken (Results 1 – 3 of 3) sorted by relevance
/external/tensorflow/tensorflow/compiler/xla/client/ |
D | xla_builder.h | 868 XlaOp SendWithToken(XlaOp operand, XlaOp token, const ChannelHandle& handle); 1411 friend XlaOp SendWithToken(XlaOp operand, XlaOp token, 2461 XlaOp SendWithToken(XlaOp operand, XlaOp token, const ChannelHandle& handle);
|
D | xla_builder.cc | 3153 return SendWithToken(operand, token, handle); in Send() 3157 XlaOp XlaBuilder::SendWithToken(XlaOp operand, XlaOp token, in SendWithToken() function in xla::XlaBuilder 4900 XlaOp SendWithToken(const XlaOp operand, const XlaOp token, in SendWithToken() function 4902 return operand.builder()->SendWithToken(operand, token, handle); in SendWithToken()
|
/external/tensorflow/tensorflow/compiler/mlir/xla/ |
D | mlir_hlo_to_hlo.cc | 995 value_map[op] = xla::SendWithToken(operand, token, in ExportXlaOp()
|