Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/core/distributed_runtime/eager/
Dremote_mgr_test.cc74 const uint64 op_id = 2; in TEST_F() local
93 const uint64 op_id = 3; in TEST_F() local
113 const uint64 op_id = 2; in TEST_F() local
134 const uint64 op_id = 3; in TEST_F() local
Dremote_tensor_handle_data.cc29 uint64 context_id, uint64 op_id, int output_num, in DestroyRemoteTensorHandle()
87 RemoteTensorHandleData::RemoteTensorHandleData(int64 op_id, int output_num, in RemoteTensorHandleData()
100 RemoteTensorHandleData::RemoteTensorHandleData(int64 op_id, int output_num, in RemoteTensorHandleData()
207 int64* op_id, in OpIdAndOutputNum()
Dremote_tensor_handle.h30 int64 op_id; member
Dremote_mgr.cc85 int64* op_id, int32* output_num) { in GetRemoteTensorHandle()
127 int64 op_id; in SerializeRemoteTensorHandle() local
Deager_service_impl_test.cc809 const int64 op_id) { in CheckOutputsAndClose()
862 const uint64 op_id = 2; in TEST_F() local
951 const int64 op_id = 2; in TEST_F() local
999 const int64 op_id = 2; in TEST_F() local
1196 int64 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/angle/third_party/vulkan-deps/spirv-tools/src/source/opt/
Dccp_pass.cpp151 if (!instr->WhileEachInId([this](uint32_t* op_id) { in VisitAssignment()
161 if (!instr->WhileEachInId([this](uint32_t* op_id) { in VisitAssignment()
Dinstruction.cpp680 ForEachInOperand([&is_opaque, this](const uint32_t* op_id) { in IsOpaqueType()
715 return WhileEachInOperand([&folder, this](const uint32_t* op_id) { in IsFoldableByFoldScalar()
/external/deqp-deps/SPIRV-Tools/source/opt/
Dccp_pass.cpp151 if (!instr->WhileEachInId([this](uint32_t* op_id) { in VisitAssignment()
161 if (!instr->WhileEachInId([this](uint32_t* op_id) { in VisitAssignment()
Dinstruction.cpp635 ForEachInOperand([&is_opaque, this](const uint32_t* op_id) { in IsOpaqueType()
670 return WhileEachInOperand([&folder, this](const uint32_t* op_id) { in IsFoldableByFoldScalar()
/external/swiftshader/third_party/SPIRV-Tools/source/opt/
Dccp_pass.cpp151 if (!instr->WhileEachInId([this](uint32_t* op_id) { in VisitAssignment()
161 if (!instr->WhileEachInId([this](uint32_t* op_id) { in VisitAssignment()
Dinstruction.cpp635 ForEachInOperand([&is_opaque, this](const uint32_t* op_id) { in IsOpaqueType()
670 return WhileEachInOperand([&folder, this](const uint32_t* op_id) { in IsFoldableByFoldScalar()
/external/tensorflow/tensorflow/core/common_runtime/eager/
Dtensor_handle.cc369 int64 op_id, int32 output_num, const string& remote_task, in CreateUnshapedRemoteHandle()
376 TensorHandle::TensorHandle(int64 op_id, int32 output_num, in TensorHandle()
396 int64 op_id, int32 output_num, tensorflow::DataType dtype, Device* d, in CreateLazyRemoteHandle()
401 TensorHandle::TensorHandle(int64 op_id, int32 output_num, in TensorHandle()
710 int64* op_id, int32* output_num) const { in RemoteAddress()
769 Status TensorHandle::AddUnshapedRemoteMirror(const Device* d, int64 op_id, in AddUnshapedRemoteMirror()
794 Status TensorHandle::AddResourceShapeMirror(const Device* d, int64 op_id, in AddResourceShapeMirror()
Dkernel_and_device.h57 int64 op_id; member
/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
630 auto op_id = in UnloadProgram() local
662 auto op_id = static_cast<PodLoadedProgramHandle*>(program)->operation_id(); in ExecuteProgram() local
795 xla::Status GetCompiledProgramShape(int64_t op_id, in GetCompiledProgramShape()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dnccl_collective_thunk.h73 int64 op_id; member
/external/tensorflow/tensorflow/c/eager/
Dtape.h421 const int64 op_id = next_op_id_++; in RecordOperation() local
452 const int64 op_id = tensor_op_it->second; in DeleteTrace() local
540 int64 op_id = op_id_it->second; in PrepareBackprop() local
847 const int64 op_id = tape_it->second; in ComputeGradient() local
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_runtime.cc598 xla::int64 op_id) { in GetRendezvousKey()
620 xla::int64 op_id, const void* replica_groups_str, in __xla_cpu_runtime_AllToAll()
662 xla::int32 channel_id_present, xla::int64 op_id, xla::int32 reduction_kind, in __xla_cpu_runtime_AllReduce()
721 xla::int64 op_id, xla::int32 byte_size, void* input_buffer, in __xla_cpu_runtime_CollectivePermute()
/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
Dcollective_ops_utils.h119 int64 op_id; member
/external/tensorflow/tensorflow/lite/toco/
Ddump_graphviz.cc524 std::string op_id = GetOpId(op_index); in DumpOperator() local
533 std::string op_id = GetOpId(op_index); in DumpOperatorEdges() local
/external/python/pybind11/include/pybind11/
Doperators.h25 enum op_id : int { enum
/external/tensorflow/tensorflow/lite/micro/
Dtest_helpers.cc262 const int op_id = in BuildSimpleStatefulModel() local
303 const int op_id = in BuildSimpleModelWithBranch() local
325 const int op_id = in BuildModelWithOfflinePlanning() local

12