Home
last modified time | relevance | path

Searched refs:op_id (Results 1 – 25 of 60) sorted by relevance

123

/external/tensorflow/tensorflow/core/distributed_runtime/eager/
Dremote_tensor_handle.h27 : 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()
Dremote_mgr_test.cc74 const uint64 op_id = 2; in TEST_F() local
76 TF_ASSERT_OK(handle->AddUnshapedRemoteMirror(remote_device_, op_id, in TEST_F()
84 EXPECT_EQ(op_id, remote_handle.op_id()); in TEST_F()
93 const uint64 op_id = 3; in TEST_F() local
96 op_id, output_num, DT_FLOAT, remote_device_, /*is_ready=*/true, ctx_); in TEST_F()
101 EXPECT_EQ(op_id, remote_handle.op_id()); in TEST_F()
113 const uint64 op_id = 2; in TEST_F() local
115 TF_ASSERT_OK(handle->AddUnshapedRemoteMirror(remote_device_, op_id, in TEST_F()
134 const uint64 op_id = 3; in TEST_F() local
137 op_id, output_num, in TEST_F()
[all …]
Dremote_mgr.cc52 remote_handle.op_id, ", Output num: ", remote_handle.output_num); in GetTensorHandleImpl()
75 remote_handle.op_id, ", Output num: ", remote_handle.output_num); in GetMirroredResourceShape()
85 int64* op_id, int32* output_num) { in GetRemoteTensorHandle() argument
87 op_id, output_num)); in GetRemoteTensorHandle()
90 GetTensorHandleImpl(RemoteTensorHandleInternal(*op_id, *output_num), &h)); in GetRemoteTensorHandle()
93 "Found two different tensor handles with the same op_id:", *op_id, in GetRemoteTensorHandle()
120 remote_handle.op_id, ", Output num: ", remote_handle.output_num); in DeleteTensorHandle()
127 int64 op_id; in SerializeRemoteTensorHandle() local
129 if (!in->RemoteAddress(device, wait_until_ready, &op_id, &output_num).ok()) { in SerializeRemoteTensorHandle()
132 GetRemoteTensorHandle(in, wait_until_ready, &op_id, &output_num)); in SerializeRemoteTensorHandle()
[all …]
Dremote_tensor_handle_data.cc29 uint64 context_id, uint64 op_id, int output_num, in DestroyRemoteTensorHandle() argument
51 handle_to_decref->set_op_id(op_id); in DestroyRemoteTensorHandle()
87 RemoteTensorHandleData::RemoteTensorHandleData(int64 op_id, int output_num, in RemoteTensorHandleData() argument
91 op_id_(op_id), in RemoteTensorHandleData()
96 << "Op ID and output num should be >= 0. Op ID: " << op_id in RemoteTensorHandleData()
100 RemoteTensorHandleData::RemoteTensorHandleData(int64 op_id, int output_num, in RemoteTensorHandleData() argument
104 op_id_(op_id), in RemoteTensorHandleData()
111 << "Op ID and output num should be >= 0. Op ID: " << op_id in RemoteTensorHandleData()
207 int64* op_id, in OpIdAndOutputNum() argument
212 *op_id = op_id_; in OpIdAndOutputNum()
Dremote_tensor_handle_data.h35 RemoteTensorHandleData(int64 op_id, int output_num, uint64 context_view_id,
39 RemoteTensorHandleData(int64 op_id, int output_num, const string& remote_task,
63 Status OpIdAndOutputNum(const bool wait_util_ready, int64* op_id,
Deager_service_impl_test.cc809 const int64 op_id) { in CheckOutputsAndClose() argument
862 const uint64 op_id = 2; in TEST_F() local
863 opts.op_id = op_id; in TEST_F()
887 CheckOutputsAndClose(outputs, op_id); in TEST_F()
951 const int64 op_id = 2; in TEST_F() local
958 [=]() { return op_id; })); in TEST_F()
985 CheckOutputsAndClose(outputs, op_id); in TEST_F()
999 const int64 op_id = 2; in TEST_F() local
1006 [=]() { return op_id; })); in TEST_F()
1039 CheckOutputsAndClose(outputs, op_id); in TEST_F()
[all …]
Deager_service_impl.cc173 EagerContext* eager_context, int op_id, int num_retvals, in AddOpRetvalsToResponse() argument
178 if (op_id == kInvalidRemoteOpId) { in AddOpRetvalsToResponse()
196 eager_context->RemoteMgr()->AddOperationOutput(retvals[i], op_id, in AddOpRetvalsToResponse()
486 [op, op_id = operation.id(), num_retvals, retvals, output_nums, cm, in RunComponentFunction()
504 eager_context, op_id, *num_retvals, output_nums, retvals->data(), in RunComponentFunction()
697 eager_context->RemoteMgr()->AddOperationOutputs(tensors, send_tensor.op_id()); in SendTensor()
744 send_packed_handle.op_id()); in SendPackedHandle()
/external/tensorflow/tensorflow/core/profiler/convert/
Dop_stack.h32 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/compiler/xla/service/
Dcollective_ops_utils.h84 CollectiveOpKind collective_op_kind, int64 op_id) in RendezvousKey()
89 op_id(op_id) {} in RendezvousKey()
95 static_cast<int>(k.collective_op_kind), k.op_id); in AbslHashValue()
101 a.op_id == b.op_id;
112 num_local_participants, static_cast<int>(collective_op_kind), op_id); in ToString()
119 int64 op_id; member
Dloop_schedule_linearizer.cc35 for (int op_id = 0; op_id < instr->operand_count(); op_id++) { in ComputationInstructionOrdering() local
36 const HloInstruction* op = instr->operand(op_id); in ComputationInstructionOrdering()
/external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/
Dpod_tpu_driver.cc442 auto op_id = static_cast<PodBufferHandle*>(handle.get())->operation_id(); in Deallocate() local
447 [this, operation_id, op_id, in Deallocate()
449 CHECK_EXISTS_OR_RETURN(underlying_buffers_, op_id, operation_id); in Deallocate()
451 auto buf_iter = underlying_buffers_.find(op_id); in Deallocate()
470 auto op_id = static_cast<PodBufferHandle*>(dst)->operation_id(); in TransferToDevice() local
475 [this, src, operation_id, op_id, in TransferToDevice()
477 CHECK_EXISTS_OR_RETURN(underlying_buffers_, op_id, operation_id); in TransferToDevice()
479 auto buf_iter = underlying_buffers_.find(op_id); in TransferToDevice()
495 auto op_id = static_cast<const PodBufferHandle*>(src)->operation_id(); in TransferFromDevice() local
500 [this, dst, operation_id, op_id, in TransferFromDevice()
[all …]
/external/tensorflow/tensorflow/core/common_runtime/eager/
Dtensor_handle.h67 TensorHandle(int64 op_id, int32 output_num, const string& remote_task,
70 TensorHandle(int64 op_id, int32 output_num, tensorflow::DataType dtype,
109 int64 op_id, int32 output_num, const string& remote_task,
116 static TensorHandle* CreateLazyRemoteHandle(int64 op_id, int32 output_num,
182 Status AddUnshapedRemoteMirror(const Device* d, int64 op_id, int output_num,
184 Status AddResourceShapeMirror(const Device* d, int64 op_id, int output_num,
191 int64* op_id, int32* output_num) const;
Dtensor_handle.cc369 int64 op_id, int32 output_num, const string& remote_task, in CreateUnshapedRemoteHandle() argument
372 return new TensorHandle(op_id, output_num, remote_task, dtype, d, ctx, in CreateUnshapedRemoteHandle()
376 TensorHandle::TensorHandle(int64 op_id, int32 output_num, in TensorHandle() argument
389 data_(absl::in_place_type<RemoteTensorHandleData>, op_id, output_num, in TensorHandle()
396 int64 op_id, int32 output_num, tensorflow::DataType dtype, Device* d, in CreateLazyRemoteHandle() argument
398 return new TensorHandle(op_id, output_num, dtype, d, is_ready, ctx); in CreateLazyRemoteHandle()
401 TensorHandle::TensorHandle(int64 op_id, int32 output_num, in TensorHandle() argument
412 data_(absl::in_place_type<RemoteTensorHandleData>, op_id, output_num, in TensorHandle()
710 int64* op_id, int32* output_num) const { in RemoteAddress() argument
718 return mirror->second.OpIdAndOutputNum(wait_until_ready, op_id, in RemoteAddress()
[all …]
Dexecute_node_test.cc126 EXPECT_EQ(remote0.op_id(), 1); in TEST()
130 EXPECT_EQ(remote1.op_id(), 2); in TEST()
/external/tensorflow/tensorflow/python/tpu/
Dtensor_tracer.py999 def _skip_op(self, op_id, op, ops_in_exec_path, report_handler): argument
1012 op, TensorTracer.reason(op_id, _REASON_WHILELOOP_OP))
1016 op, TensorTracer.reason(op_id, _REASON_CONTROLFLOW_OP))
1020 op, TensorTracer.reason(op_id, _REASON_UNSAFE_OP))
1024 op, TensorTracer.reason(op_id, _REASON_DEVICE_MISMATCH))
1028 op, TensorTracer.reason(op_id, _REASON_NOT_EXECUTED))
1044 op, TensorTracer.reason(op_id, _REASON_IN_CONTROL_FLOW))
1048 op, TensorTracer.reason(op_id, _REASON_USER_INCLUDED))
1051 if not self._inside_op_range(op_id):
1053 op, TensorTracer.reason(op_id, _REASON_OUTSIDE_OP_RANGE))
[all …]
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dnccl_collective_thunk.h73 int64 op_id; member
93 config.op_id = op.channel_id()->handle().getUInt(); in GetNcclCollectiveConfigForMlir()
99 config.op_id = static_cast<int64>(unique_id.getInt()); in GetNcclCollectiveConfigForMlir()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_runtime.h176 xla::int32 channel_id_present, xla::int64 op_id, xla::int32 reduction_kind,
182 xla::int64 op_id, xla::int32 byte_size, void* input_buffer,
188 xla::int64 op_id, const void* replica_groups_str,
Dcpu_runtime.cc598 xla::int64 op_id) { in GetRendezvousKey() argument
613 num_local_participants, op_kind, op_id}; in GetRendezvousKey()
620 xla::int64 op_id, const void* replica_groups_str, in __xla_cpu_runtime_AllToAll() argument
633 GetRendezvousKey(run_options, group, channel_id_present, op_id); in __xla_cpu_runtime_AllToAll()
662 xla::int32 channel_id_present, xla::int64 op_id, xla::int32 reduction_kind, in __xla_cpu_runtime_AllReduce() argument
671 GetRendezvousKey(run_options, group, channel_id_present, op_id); in __xla_cpu_runtime_AllReduce()
721 xla::int64 op_id, xla::int32 byte_size, void* input_buffer, in __xla_cpu_runtime_CollectivePermute() argument
743 GetRendezvousKey(run_options, {}, channel_id_present, op_id); in __xla_cpu_runtime_CollectivePermute()
/external/tensorflow/tensorflow/c/eager/
Dtape.h421 const int64 op_id = next_op_id_++; in RecordOperation() local
427 tensor_tape_[o.GetID()] = op_id; in RecordOperation()
431 op_tape_[op_id] = OpTapeEntry<BackwardFunction, TapeTensor>{ in RecordOperation()
452 const int64 op_id = tensor_op_it->second; in DeleteTrace() local
453 if (op_id == -1) { in DeleteTrace()
458 auto op_it = op_tape_.find(op_id); in DeleteTrace()
540 int64 op_id = op_id_it->second; in PrepareBackprop() local
541 auto op_it = op_tape->find(op_id); in PrepareBackprop()
542 auto result_op_it = result.op_tape.find(op_id); in PrepareBackprop()
543 if (op_id == -1 || op_it == op_tape->end() || in PrepareBackprop()
[all …]
/external/rust/crates/grpcio-sys/grpc/src/core/ext/transport/cronet/transport/
Dcronet_transport.cc825 enum e_op_id op_id) { in op_can_be_run() argument
835 if (op_id == OP_SEND_INITIAL_METADATA) { in op_can_be_run()
839 if (op_id == OP_SEND_MESSAGE) { in op_can_be_run()
843 if (op_id == OP_SEND_TRAILING_METADATA) { in op_can_be_run()
847 if (op_id == OP_CANCEL_ERROR) { in op_can_be_run()
852 if (op_id == OP_RECV_INITIAL_METADATA && in op_can_be_run()
857 if (op_id == OP_RECV_MESSAGE && op_state->state_op_done[OP_RECV_MESSAGE]) { in op_can_be_run()
861 if (op_id == OP_RECV_TRAILING_METADATA && in op_can_be_run()
871 if (op_id == OP_ON_COMPLETE && in op_can_be_run()
879 } else if (op_id == OP_SEND_INITIAL_METADATA) { in op_can_be_run()
[all …]
/external/grpc-grpc/src/core/ext/transport/cronet/transport/
Dcronet_transport.cc807 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/angle/third_party/vulkan-deps/spirv-tools/src/source/opt/
Dccp_pass.cpp151 if (!instr->WhileEachInId([this](uint32_t* op_id) { in VisitAssignment() argument
152 auto iter = values_.find(*op_id); in VisitAssignment()
161 if (!instr->WhileEachInId([this](uint32_t* op_id) { in VisitAssignment() argument
162 auto it = values_.find(*op_id); in VisitAssignment()
/external/deqp-deps/SPIRV-Tools/source/opt/
Dccp_pass.cpp151 if (!instr->WhileEachInId([this](uint32_t* op_id) { in VisitAssignment() argument
152 auto iter = values_.find(*op_id); in VisitAssignment()
161 if (!instr->WhileEachInId([this](uint32_t* op_id) { in VisitAssignment() argument
162 auto it = values_.find(*op_id); in VisitAssignment()
/external/swiftshader/third_party/SPIRV-Tools/source/opt/
Dccp_pass.cpp151 if (!instr->WhileEachInId([this](uint32_t* op_id) { in VisitAssignment() argument
152 auto iter = values_.find(*op_id); in VisitAssignment()
161 if (!instr->WhileEachInId([this](uint32_t* op_id) { in VisitAssignment() argument
162 auto it = values_.find(*op_id); in VisitAssignment()
/external/python/cpython3/Tools/c-analyzer/
DTODO229 Objects/typeobject.c:slot_nb_add():op_id _Py_static_string(op_id, OPSTR)
230 Objects/typeobject.c:slot_nb_add():rop_id _Py_static_string(op_id, OPSTR)
231 Objects/typeobject.c:slot_nb_and():op_id _Py_static_string(op_id, OPSTR)
232 Objects/typeobject.c:slot_nb_and():rop_id _Py_static_string(op_id, OPSTR)
233 Objects/typeobject.c:slot_nb_divmod():op_id _Py_static_string(op_id, OPSTR)
234 Objects/typeobject.c:slot_nb_divmod():rop_id _Py_static_string(op_id, OPSTR)
235 Objects/typeobject.c:slot_nb_floor_divide():op_id _Py_static_string(op_id, OPSTR)
236 Objects/typeobject.c:slot_nb_floor_divide():rop_id _Py_static_string(op_id, OPSTR)
237 Objects/typeobject.c:slot_nb_lshift():op_id _Py_static_string(op_id, OPSTR)
238 Objects/typeobject.c:slot_nb_lshift():rop_id _Py_static_string(op_id, OPSTR)
[all …]

123