| /external/tensorflow/tensorflow/core/distributed_runtime/eager/ |
| D | remote_tensor_handle.h | 27 : op_id(tensor_handle.op_id()), output_num(tensor_handle.output_num()) {} in RemoteTensorHandleInternal()
28 RemoteTensorHandleInternal(int64 op_id, int32 output_num) in RemoteTensorHandleInternal()
29 : op_id(op_id), output_num(output_num) {} in RemoteTensorHandleInternal()
30 int64 op_id; member
36 return FingerprintCat64(handle.op_id, handle.output_num); in operator()
43 return first.op_id == second.op_id && first.output_num == second.output_num; in operator()
|
| D | remote_mgr.cc | 45 remote_handle.op_id, ", Output num: ", remote_handle.output_num); in GetTensorHandleImpl()
68 remote_handle.op_id, ", Output num: ", remote_handle.output_num); in GetMirroredResourceShape()
77 int64* op_id, int32* output_num) { in GetRemoteTensorHandle() argument
79 handle->RemoteAddress(handle->device(), op_id, output_num)); in GetRemoteTensorHandle()
82 GetTensorHandleImpl(RemoteTensorHandleInternal(*op_id, *output_num), &h)); in GetRemoteTensorHandle()
85 "Found two different tensor handles with the same op_id:", *op_id, in GetRemoteTensorHandle()
112 remote_handle.op_id, ", Output num: ", remote_handle.output_num); in DeleteTensorHandle()
118 int64 op_id; in SerializeRemoteTensorHandle() local
120 if (!in->RemoteAddress(device, &op_id, &output_num).ok()) { in SerializeRemoteTensorHandle()
122 TF_RETURN_IF_ERROR(GetRemoteTensorHandle(in, &op_id, &output_num)); in SerializeRemoteTensorHandle()
[all …]
|
| D | remote_mgr_test.cc | 78 const uint64 op_id = 2; in TEST_F() local
81 op_id, output_num, t.shape(), /*remote_task=*/"", context_id_, ctx_); in TEST_F()
87 EXPECT_EQ(op_id, remote_handle.op_id()); in TEST_F()
97 const uint64 op_id = 3; in TEST_F() local
101 op_id, output_num, t.shape(), /*remote_task=*/"", context_id_, DT_FLOAT, in TEST_F()
107 EXPECT_EQ(op_id, remote_handle.op_id()); in TEST_F()
|
| D | remote_tensor_handle_data.cc | 28 uint64 context_id, uint64 op_id, int output_num, in DestroyRemoteTensorHandle() argument
50 handle_to_decref->set_op_id(op_id); in DestroyRemoteTensorHandle()
86 RemoteTensorHandleData::RemoteTensorHandleData(int64 op_id, int output_num, in RemoteTensorHandleData() argument
91 : op_id_(op_id), in RemoteTensorHandleData()
98 << "Op ID and output num should be >= 0. Op ID: " << op_id in RemoteTensorHandleData()
151 int64 op_id, int32 output_num, const string& remote_task, uint64 context_id, in UnshapedRemoteTensorHandleData() argument
153 : op_id_(op_id), in UnshapedRemoteTensorHandleData()
160 << "Op ID and output num should be >= 0. Op ID: " << op_id in UnshapedRemoteTensorHandleData()
|
| D | remote_tensor_handle_data.h | 27 RemoteTensorHandleData(int64 op_id, int output_num, const TensorShape& shape,
43 int64 op_id() const { return op_id_; } in op_id() function
60 UnshapedRemoteTensorHandleData(int64 op_id, int32 output_num,
76 int64 op_id() const { return op_id_; } in op_id() function
|
| D | eager_service_impl_test.cc | 517 void CheckOutputsAndClose(const int64 op_id) { in CheckOutputsAndClose() argument
570 const uint64 op_id = 2; in TEST_F() local
571 opts.op_id = op_id; in TEST_F()
595 CheckOutputsAndClose(op_id); in TEST_F()
609 const int64 op_id = 2; in TEST_F() local
614 [=]() { return op_id; })); in TEST_F()
639 CheckOutputsAndClose(op_id); in TEST_F()
|
| D | eager_service_impl.h | 181 << handle_to_delete_->op_id << ":" in Run()
191 strings::StrAppend(&out, " op_id: ", handle_to_delete_->op_id); in DebugString()
|
| D | cluster_function_library_runtime.cc | 139 if (!opts.op_id.has_value()) { in Run()
153 remote_op->set_id(opts.op_id.value()); in Run()
|
| D | remote_execute_node.cc | 43 item.handle_to_decref().op_id(), ":", in RunAsync()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | collective_ops_utils.h | 76 CollectiveOpKind collective_op_kind, int64 op_id) in RendezvousKey()
80 op_id(op_id) {} in RendezvousKey()
86 int64 op_id; in FromInstruction() local
88 std::tie(collective_op_kind, op_id) = in FromInstruction()
95 op_id); in FromInstruction()
103 static_cast<int>(k.collective_op_kind), k.op_id); in AbslHashValue()
109 a.op_id == b.op_id;
120 static_cast<int>(collective_op_kind), op_id); in ToString()
126 int64 op_id; member
|
| /external/tensorflow/tensorflow/core/profiler/convert/ |
| D | op_stack.h | 32 void Push(uint32 op_id, std::unique_ptr<OpInfo> op_info) { in Push() argument
33 stack_.emplace_back(op_id, std::move(op_info)); in Push()
37 std::unique_ptr<OpInfo> Pop(uint32 op_id) { in Pop() argument
43 if (op_id == back.first) { in Pop()
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | tensor_handle.cc | 202 Status TensorHandle::CreateRemoteHandle(int64 op_id, int output_num, in CreateRemoteHandle() argument
209 absl::make_unique<RemoteTensorHandleData>(op_id, output_num, shape, in CreateRemoteHandle()
222 remote_op_id_(t->op_id()), in TensorHandle()
242 int64 op_id, int32 output_num, const string& remote_task, uint64 context_id, in CreateUnshapedRemoteHandle() argument
245 op_id, output_num, remote_task, context_id, ctx), in CreateUnshapedRemoteHandle()
256 remote_op_id_(t->op_id()), in TensorHandle()
414 Status TensorHandle::RemoteAddress(Device* d, int64* op_id, in RemoteAddress() argument
420 *op_id = mirror->second->op_id(); in RemoteAddress()
427 *op_id = unshaped_mirror->second->op_id(); in RemoteAddress()
442 *op_id = remote_op_id_; in RemoteAddress()
[all …]
|
| D | tensor_handle.h | 94 static Status CreateRemoteHandle(int64 op_id, int output_num,
104 static Status CreateUnshapedRemoteHandle(int64 op_id, int32 output_num,
142 Status RemoteAddress(Device* d, int64* op_id, int32* output_num) const;
146 void SetRemoteOpIdAndOutputNumToLocalTensorHandle(const int64 op_id,
|
| D | kernel_and_device.cc | 322 opts->op_id = params.op_id; in Run()
328 opts->op_id = get_op_id_(); in Run()
|
| /external/tensorflow/tensorflow/python/tpu/ |
| D | tensor_tracer.py | 860 def _skip_op(self, op_id, op, ops_in_exec_path, report_handler): argument
873 op, TensorTracer.reason(op_id, _REASON_WHILELOOP_OP))
877 op, TensorTracer.reason(op_id, _REASON_CONTROLFLOW_OP))
881 op, TensorTracer.reason(op_id, _REASON_UNSAFE_OP))
885 op, TensorTracer.reason(op_id, _REASON_DEVICE_MISMATCH))
889 op, TensorTracer.reason(op_id, _REASON_NOT_EXECUTED))
894 op, TensorTracer.reason(op_id, _REASON_USER_INCLUDED))
897 if not self._inside_op_range(op_id):
899 op, TensorTracer.reason(op_id, _REASON_OUTSIDE_OP_RANGE))
903 op, TensorTracer.reason(op_id, _REASON_LESS_INTERESTING_OP))
[all …]
|
| /external/tensorflow/tensorflow/c/eager/ |
| D | tape.h | 393 const int64 op_id = next_op_id_++; in RecordOperation() local
399 tensor_tape_[o.GetID()] = op_id; in RecordOperation()
403 op_tape_[op_id] = OpTapeEntry<BackwardFunction, TapeTensor>{ in RecordOperation()
424 const int64 op_id = tensor_op_it->second; in DeleteTrace() local
425 if (op_id == -1) { in DeleteTrace()
430 auto op_it = op_tape_.find(op_id); in DeleteTrace()
512 int64 op_id = op_id_it->second; in PrepareBackprop() local
513 auto op_it = op_tape->find(op_id); in PrepareBackprop()
514 auto result_op_it = result.op_tape.find(op_id); in PrepareBackprop()
515 if (op_id == -1 || op_it == op_tape->end() || in PrepareBackprop()
[all …]
|
| /external/swiftshader/third_party/SPIRV-Tools/source/opt/ |
| D | ccp_pass.cpp | 150 if (!instr->WhileEachInId([this](uint32_t* op_id) { in VisitAssignment() argument
151 auto iter = values_.find(*op_id); in VisitAssignment()
160 if (!instr->WhileEachInId([this](uint32_t* op_id) { in VisitAssignment() argument
161 auto it = values_.find(*op_id); in VisitAssignment()
|
| /external/angle/third_party/spirv-tools/src/source/opt/ |
| D | ccp_pass.cpp | 150 if (!instr->WhileEachInId([this](uint32_t* op_id) { in VisitAssignment() argument
151 auto iter = values_.find(*op_id); in VisitAssignment()
160 if (!instr->WhileEachInId([this](uint32_t* op_id) { in VisitAssignment() argument
161 auto it = values_.find(*op_id); in VisitAssignment()
|
| /external/deqp-deps/SPIRV-Tools/source/opt/ |
| D | ccp_pass.cpp | 150 if (!instr->WhileEachInId([this](uint32_t* op_id) { in VisitAssignment() argument
151 auto iter = values_.find(*op_id); in VisitAssignment()
160 if (!instr->WhileEachInId([this](uint32_t* op_id) { in VisitAssignment() argument
161 auto it = values_.find(*op_id); in VisitAssignment()
|
| /external/grpc-grpc/src/core/ext/transport/cronet/transport/ |
| D | cronet_transport.cc | 807 enum e_op_id op_id) { in op_can_be_run() argument
817 if (op_id == OP_SEND_INITIAL_METADATA) { in op_can_be_run()
821 if (op_id == OP_SEND_MESSAGE) { in op_can_be_run()
825 if (op_id == OP_SEND_TRAILING_METADATA) { in op_can_be_run()
829 if (op_id == OP_CANCEL_ERROR) { in op_can_be_run()
834 if (op_id == OP_RECV_INITIAL_METADATA && in op_can_be_run()
839 if (op_id == OP_RECV_MESSAGE && op_state->state_op_done[OP_RECV_MESSAGE]) { in op_can_be_run()
843 if (op_id == OP_RECV_TRAILING_METADATA && in op_can_be_run()
853 if (op_id == OP_ON_COMPLETE && in op_can_be_run()
861 } else if (op_id == OP_SEND_INITIAL_METADATA) { in op_can_be_run()
[all …]
|
| /external/autotest/client/cros/cellular/pseudomodem/ |
| D | sim.py | 258 op_id = ''
262 op_id = self._carrier.operator_id
267 'OperatorIdentifier' : op_id,
|
| /external/python/pybind11/include/pybind11/ |
| D | operators.h | 25 enum op_id : int { enum
50 template <op_id, op_type, typename B, typename L, typename R> struct op_impl { };
53 template <op_id id, op_type ot, typename L, typename R> struct op_ {
|
| /external/tensorflow/tensorflow/lite/toco/ |
| D | dump_graphviz.cc | 520 string op_id = GetOpId(op_index); in DumpOperator() local
521 AppendF(output_file, kOpNodeFmt, op_id, label, color.AsHexString(), in DumpOperator()
528 string op_id = GetOpId(op_index); in DumpOperatorEdges() local
550 AppendF(output_file, kInputEdgeFmt, input, compass_pt, op_id, i, line_width, in DumpOperatorEdges()
567 AppendF(output_file, kOutputEdgeFmt, op_id, i, output, compass_pt, in DumpOperatorEdges()
|
| /external/tensorflow/tensorflow/core/kernels/hexagon/ |
| D | soc_interface.h | 93 bool soc_interface_AppendNode(const char* const name, int node_id, int op_id,
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_runtime.h | 169 xla::int32 channel_id_present, xla::int64 op_id, xla::int32 reduction_kind,
|