Searched refs:send_op (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | rewrite_tpu_embedding_ops.cc | 67 SendTPUEmbeddingGradientsOp send_op; in runOnFunction() local
68 if (failed(GetOp(func, &send_op))) return signalPassFailure(); in runOnFunction()
71 if (!recv_op && !send_op) return; in runOnFunction()
73 Location loc = recv_op ? recv_op.getLoc() : send_op.getLoc(); in runOnFunction()
74 StringRef config = recv_op ? recv_op.config() : send_op.config(); in runOnFunction()
89 if (send_op) { in runOnFunction()
90 int32_t operand_sizes[] = {static_cast<int32_t>(send_op.N()), in runOnFunction()
91 static_cast<int32_t>(send_op.NN()), 1}; in runOnFunction()
96 send_op, dedup_op, &builder); in runOnFunction()
|
| D | cross_host_transfer.cc | 131 auto send_op = in runOnFunction() local
133 send_op->setAttr(kOpDeviceAttr, in runOnFunction()
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | auto_control_deps_test.py | 172 send_op = gen_sendrecv_ops.send(v, "x", "/", 0, "/")
175 self.assertIn(send_op, c.ops_which_must_run)
|
| /external/tensorflow/tensorflow/core/graph/ |
| D | graph_partition.cc | 241 const string send_op = (host_memory) ? "_HostSend" : "_Send"; in AddSend() local
242 NodeDefBuilder send_builder(opts.new_name(src->name()), send_op, in AddSend()
|
| /external/mesa3d/src/intel/compiler/ |
| D | brw_eu.h | 1258 enum opcode send_op,
|
| D | brw_fs_generator.cpp | 2391 const enum opcode send_op = inst->opcode == SHADER_OPCODE_INTERLOCK ? in generate_code()
2394 brw_memory_fence(p, dst, src[0], send_op, in generate_code()
|
| D | brw_eu_emit.c | 3204 enum opcode send_op, in brw_memory_fence() argument
3217 struct brw_inst *insn = next_insn(p, send_op); in brw_memory_fence()
|