Searched refs:send_op (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/c/ |
| D | c_api_experimental_test.cc | 232 TFE_Op* send_op = TFE_NewOp(ctx, "_Send", status); in TEST() local
234 TFE_OpAddInput(send_op, m, status); in TEST()
238 TFE_OpSetAttrType(send_op, "T", TF_FLOAT); in TEST()
239 TFE_OpSetAttrString(send_op, "tensor_name", tensor_name.c_str(), in TEST()
242 TFE_OpSetAttrString(send_op, "send_device", send_device.c_str(), in TEST()
244 TFE_OpSetAttrInt(send_op, "send_device_incarnation", 1234); in TEST()
246 TFE_OpSetAttrString(send_op, "recv_device", recv_device.c_str(), in TEST()
248 TFE_OpSetAttrBool(send_op, "client_terminated", true); in TEST()
266 auto* send_result = TFE_ExecuteOpInNewThread(send_op, &send_retvals, in TEST()
291 TFE_DeleteOp(send_op); in TEST()
|
| /external/tensorflow/tensorflow/core/graph/ |
| D | graph_partition.cc | 237 const string send_op = (host_memory) ? "_HostSend" : "_Send"; in AddSend() local
238 NodeDefBuilder send_builder(opts.new_name(src->name()), send_op, in AddSend()
|