Home
last modified time | relevance | path

Searched defs:op_id (Results 1 – 25 of 43) sorted by relevance

12

/external/tensorflow/tensorflow/core/distributed_runtime/eager/
Dremote_mgr_test.cc77 const uint64 op_id = 2; in TEST_F() local
96 const uint64 op_id = 3; in TEST_F() local
116 const uint64 op_id = 2; in TEST_F() local
137 const uint64 op_id = 3; in TEST_F() local
160 const uint64 op_id = 3; in TEST_F() local
Dremote_tensor_handle_data.cc30 uint64 context_id, uint64 op_id, int output_num, in DestroyRemoteTensorHandle()
88 RemoteTensorHandleData::RemoteTensorHandleData(int64_t op_id, int output_num, in RemoteTensorHandleData()
101 RemoteTensorHandleData::RemoteTensorHandleData(int64_t op_id, int output_num, in RemoteTensorHandleData()
208 int64_t* op_id, in OpIdAndOutputNum()
Dremote_tensor_handle.h30 int64_t op_id; member
Dremote_mgr.cc113 int64_t* op_id, int32* output_num) { in GetRemoteTensorHandle()
155 int64_t op_id; in SerializeRemoteTensorHandle() local
Deager_service_impl_test.cc815 const int64_t op_id) { in CheckOutputsAndClose()
868 const uint64 op_id = 2; in TEST_F() local
957 const int64_t op_id = 2; in TEST_F() local
1011 const int64_t op_id = 2; in TEST_F() local
1214 int64_t op_id; in TEST_F() local
/external/tensorflow/tensorflow/core/profiler/convert/
Dop_stack.h32 void Push(uint32 op_id, std::unique_ptr<OpInfo> op_info) { in Push()
37 std::unique_ptr<OpInfo> Pop(uint32 op_id) { in Pop()
/external/tensorflow/tensorflow/core/protobuf/
Dremote_tensor_handle.proto21 int64 op_id = 1; field
Deager_service.proto171 repeated int64 op_id = 2; field
238 int64 op_id = 1; field
250 int64 op_id = 1; field
/external/swiftshader/third_party/SPIRV-Tools/source/opt/
Dccp_pass.cpp182 if (!instr->WhileEachInId([this](uint32_t* op_id) { in VisitAssignment()
192 if (!instr->WhileEachInId([this](uint32_t* op_id) { in VisitAssignment()
Dinstruction.cpp737 ForEachInOperand([&is_opaque, this](const uint32_t* op_id) { in IsOpaqueType()
772 return WhileEachInOperand([&folder, this](const uint32_t* op_id) { in IsFoldableByFoldScalar()
/external/deqp-deps/SPIRV-Tools/source/opt/
Dccp_pass.cpp185 if (!instr->WhileEachInId([this](uint32_t* op_id) { in VisitAssignment()
195 if (!instr->WhileEachInId([this](uint32_t* op_id) { in VisitAssignment()
Dinstruction.cpp722 ForEachInOperand([&is_opaque, this](const uint32_t* op_id) { in IsOpaqueType()
757 return WhileEachInOperand([&folder, this](const uint32_t* op_id) { in IsFoldableByFoldScalar()
/external/angle/third_party/vulkan-deps/spirv-tools/src/source/opt/
Dccp_pass.cpp181 if (!instr->WhileEachInId([this](uint32_t* op_id) { in VisitAssignment()
191 if (!instr->WhileEachInId([this](uint32_t* op_id) { in VisitAssignment()
Dinstruction.cpp737 ForEachInOperand([&is_opaque, this](const uint32_t* op_id) { in IsOpaqueType()
772 return WhileEachInOperand([&folder, this](const uint32_t* op_id) { in IsFoldableByFoldScalar()
794 return WhileEachInOperand([&folder, this](const uint32_t* op_id) { in IsFoldableByFoldVector()
Dtrim_capabilities_pass.cpp100 item->ForEachInOperand([&instructions_to_visit](const uint32_t* op_id) { in Handler_OpVariable_StorageInputOutput16()
/external/tensorflow/tensorflow/core/common_runtime/eager/
Dtensor_handle.cc369 int64_t op_id, int32_t output_num, const string& remote_task, in CreateUnshapedRemoteHandle()
376 TensorHandle::TensorHandle(int64_t op_id, int32_t output_num, in TensorHandle()
396 int64_t op_id, int32_t output_num, tensorflow::DataType dtype, Device* d, in CreateLazyRemoteHandle()
401 TensorHandle::TensorHandle(int64_t op_id, int32_t output_num, in TensorHandle()
710 int64_t* op_id, int32* output_num) const { in RemoteAddress()
769 Status TensorHandle::AddUnshapedRemoteMirror(const Device* d, int64_t op_id, in AddUnshapedRemoteMirror()
794 Status TensorHandle::AddResourceShapeMirror(const Device* d, int64_t op_id, in AddResourceShapeMirror()
/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
470 auto op_id = static_cast<PodBufferHandle*>(dst)->operation_id(); in TransferToDevice() local
495 auto op_id = static_cast<const PodBufferHandle*>(src)->operation_id(); in TransferFromDevice() local
631 auto op_id = in UnloadProgram() local
663 auto op_id = static_cast<PodLoadedProgramHandle*>(program)->operation_id(); in ExecuteProgram() local
799 xla::Status GetCompiledProgramShape(int64_t op_id, in GetCompiledProgramShape()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dnccl_utils.cc140 RunId run_id, OpId op_id, NcclCliqueKey clique_key, in AcquireNcclClique()
224 RunId run_id, OpId op_id, std::vector<GlobalDeviceId> participants, in AcquireNcclComm()
Dnccl_collective_thunk.cc116 CollectiveOpGroupMode group_mode, int64_t op_id) { in LockNcclComm()
Dnccl_collective_thunk.h55 int64_t op_id; member
Djitrt_custom_calls.cc345 int64_t group_mode, int64_t op_id, in GetNcclComm()
1608 int64_t group_mode, int64_t op_id, in operator ()()
1676 int64_t group_mode, int64_t op_id, in operator ()()
1805 int64_t group_mode, int64_t op_id, in operator ()()
1873 int64_t group_mode, int64_t op_id, in operator ()()
1938 int64_t op_id, in operator ()()
2005 int32_t uid, int64_t group_mode, int64_t op_id, in operator ()()
/external/tensorflow/tensorflow/c/eager/
Dtape.h421 const int64_t op_id = next_op_id_++; in RecordOperation() local
452 const int64_t op_id = tensor_op_it->second; in DeleteTrace() local
540 int64_t op_id = op_id_it->second; in PrepareBackprop() local
847 const int64_t op_id = tape_it->second; in ComputeGradient() local
/external/tensorflow/tensorflow/compiler/xla/service/
Dloop_schedule_linearizer.cc35 for (int op_id = 0; op_id < instr->operand_count(); op_id++) { in ComputationInstructionOrdering() local
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_runtime.cc696 std::optional<bool> use_global_device_ids, int64_t op_id) { in GetRendezvousKey()
720 int64_t op_id, const void* replica_groups_str, in __xla_cpu_runtime_AllToAll()
763 int32_t channel_id_present, int32_t use_global_device_ids, int64_t op_id, in __xla_cpu_runtime_AllReduce()
832 int64_t op_id, int32_t byte_size, void* input_buffer, void* output_buffer, in __xla_cpu_runtime_CollectivePermute()
/external/tensorflow/tensorflow/lite/toco/
Ddump_graphviz.cc526 std::string op_id = GetOpId(op_index); in DumpOperator() local
535 std::string op_id = GetOpId(op_index); in DumpOperatorEdges() local

12