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()
|