Home
last modified time | relevance | path

Searched refs:remote_op (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/core/distributed_runtime/eager/
Dcluster_function_library_runtime.cc146 eager::Operation* remote_op = request->add_queue()->mutable_operation(); in Run() local
148 remote_op->add_inputs()->Swap(&(*args)[i]); in Run()
153 remote_op->set_id(opts.op_id.value()); in Run()
154 remote_op->set_is_function(true); in Run()
155 remote_op->set_is_component_function(true); in Run()
156 remote_op->set_func_step_id(opts.step_id); in Run()
157 remote_op->set_name(op->Name()); in Run()
158 op->Attrs().FillAttrValueMap(remote_op->mutable_attrs()); in Run()
159 remote_op->set_device(function_data->target); in Run()
Dremote_copy_node.cc34 void PrepareRemoteOp(eager::Operation* remote_op, EagerOperation* op) { in PrepareRemoteOp() argument
35 remote_op->set_name(op->Name()); in PrepareRemoteOp()
37 op->Attrs().FillAttrValueMap(remote_op->mutable_attrs()); in PrepareRemoteOp()
38 remote_op->set_device(op->Device()->name()); in PrepareRemoteOp()
146 auto* remote_op = request.add_queue()->mutable_operation(); in StartSend() local
148 src_, remote_op->add_inputs(), src_->device(), in StartSend()
155 PrepareRemoteOp(remote_op, &op); in StartSend()
156 remote_op->set_id(ctx_->RemoteMgr()->NextOpId()); in StartSend()
198 auto* remote_op = request.add_queue()->mutable_operation(); in RunRemoteRecv() local
199 PrepareRemoteOp(remote_op, op); in RunRemoteRecv()
[all …]
/external/tensorflow/tensorflow/python/data/kernel_tests/
Diterator_cluster_test.py82 remote_op = functional_ops.remote_call(
89 elem = sess.run(remote_op, feed_dict={target_placeholder: device1})
93 sess.run(remote_op, feed_dict={target_placeholder: device0})
96 elem = sess.run(remote_op, feed_dict={target_placeholder: device1})
99 sess.run(remote_op, feed_dict={target_placeholder: device1})
Diterator_test.py618 remote_op = functional_ops.remote_call(
626 remote_op,
634 remote_op,
639 remote_op,
645 remote_op,
652 remote_op,
736 remote_op = functional_ops.remote_call(
744 remote_op,
750 remote_op,
756 remote_op,
[all …]
/external/tensorflow/tensorflow/core/common_runtime/eager/
Dexecute.cc595 void PrepareRemoteOp(eager::Operation* remote_op, EagerOperation* op) { in PrepareRemoteOp() argument
598 remote_op->set_id(ctx.RemoteMgr()->NextOpId()); in PrepareRemoteOp()
599 remote_op->set_name(op->Name()); in PrepareRemoteOp()
601 op->Attrs().FillAttrValueMapWithoutDefaults(remote_op->mutable_attrs()); in PrepareRemoteOp()
602 remote_op->set_device(op->Device()->name()); in PrepareRemoteOp()
603 remote_op->set_is_function(op->is_function()); in PrepareRemoteOp()
606 Status StoreResourceDtypesAndShapes(const eager::Operation& remote_op, in StoreResourceDtypesAndShapes() argument
609 if (remote_op.name() == "VarHandleOp") { in StoreResourceDtypesAndShapes()
617 AttrSlice attr_slice = AttrSlice(&remote_op.attrs()); in StoreResourceDtypesAndShapes()
653 eager::Operation* remote_op = request->add_queue()->mutable_operation(); in EagerRemoteExecute() local
[all …]
/external/tensorflow/tensorflow/python/kernel_tests/
Dfunctional_ops_test.py423 remote_op = functional_ops.remote_call(
431 mul = self.evaluate(remote_op)
448 remote_op = functional_ops.remote_call(
456 mul = self.evaluate(remote_op)
471 remote_op = functional_ops.remote_call(
476 mul = self.evaluate(remote_op)
493 remote_op = functional_ops.remote_call(
501 mul = self.evaluate(remote_op)
518 remote_op = functional_ops.remote_call(
526 mul = self.evaluate(remote_op)
[all …]