Home
last modified time | relevance | path

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

12

/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/distributed_runtime/eager/
Dremote_tensor_handle_data.h43 int64 op_id() const { return op_id_; } in op_id() function
76 int64 op_id() const { return op_id_; } in op_id() function
Dremote_tensor_handle_data.cc28 uint64 context_id, uint64 op_id, int output_num, in DestroyRemoteTensorHandle()
86 RemoteTensorHandleData::RemoteTensorHandleData(int64 op_id, int output_num, in RemoteTensorHandleData()
151 int64 op_id, int32 output_num, const string& remote_task, uint64 context_id, in UnshapedRemoteTensorHandleData()
Dremote_mgr_test.cc78 const uint64 op_id = 2; in TEST_F() local
97 const uint64 op_id = 3; in TEST_F() local
Dremote_tensor_handle.h30 int64 op_id; member
Dremote_mgr.cc77 int64* op_id, int32* output_num) { in GetRemoteTensorHandle()
118 int64 op_id; in SerializeRemoteTensorHandle() local
Deager_service_impl_test.cc517 void CheckOutputsAndClose(const int64 op_id) { in CheckOutputsAndClose()
570 const uint64 op_id = 2; in TEST_F() local
609 const int64 op_id = 2; in TEST_F() local
/external/tensorflow/tensorflow/core/protobuf/
Dremote_tensor_handle.proto22 int64 op_id = 1; field
Deager_service.proto149 repeated int64 op_id = 2; field
199 int64 op_id = 1; field
/external/swiftshader/third_party/SPIRV-Tools/source/opt/
Dccp_pass.cpp150 if (!instr->WhileEachInId([this](uint32_t* op_id) { in VisitAssignment()
160 if (!instr->WhileEachInId([this](uint32_t* op_id) { in VisitAssignment()
Dfold.cpp638 &id_map](uint32_t* op_id) { in FoldInstructionToConstant()
Dinstruction.cpp454 ForEachInOperand([&is_opaque, this](const uint32_t* op_id) { in IsOpaqueType()
/external/angle/third_party/spirv-tools/src/source/opt/
Dccp_pass.cpp150 if (!instr->WhileEachInId([this](uint32_t* op_id) { in VisitAssignment()
160 if (!instr->WhileEachInId([this](uint32_t* op_id) { in VisitAssignment()
Dfold.cpp638 &id_map](uint32_t* op_id) { in FoldInstructionToConstant()
/external/deqp-deps/SPIRV-Tools/source/opt/
Dccp_pass.cpp150 if (!instr->WhileEachInId([this](uint32_t* op_id) { in VisitAssignment()
160 if (!instr->WhileEachInId([this](uint32_t* op_id) { in VisitAssignment()
Dfold.cpp638 &id_map](uint32_t* op_id) { in FoldInstructionToConstant()
/external/tensorflow/tensorflow/compiler/xla/service/
Dcollective_ops_utils.h86 int64 op_id; in FromInstruction() local
126 int64 op_id; member
/external/tensorflow/tensorflow/core/common_runtime/eager/
Dtensor_handle.cc202 Status TensorHandle::CreateRemoteHandle(int64 op_id, int output_num, in CreateRemoteHandle()
242 int64 op_id, int32 output_num, const string& remote_task, uint64 context_id, in CreateUnshapedRemoteHandle()
414 Status TensorHandle::RemoteAddress(Device* d, int64* op_id, in RemoteAddress()
448 const int64 op_id, const int32 output_num) { in SetRemoteOpIdAndOutputNumToLocalTensorHandle()
Dkernel_and_device.h53 int64 op_id; member
/external/tensorflow/tensorflow/c/eager/
Dtape.h393 const int64 op_id = next_op_id_++; in RecordOperation() local
424 const int64 op_id = tensor_op_it->second; in DeleteTrace() local
512 int64 op_id = op_id_it->second; in PrepareBackprop() local
822 const int64 op_id = tape_it->second; in ComputeGradient() local
/external/tensorflow/tensorflow/lite/toco/
Ddump_graphviz.cc520 string op_id = GetOpId(op_index); in DumpOperator() local
528 string op_id = GetOpId(op_index); in DumpOperatorEdges() local
/external/python/pybind11/include/pybind11/
Doperators.h25 enum op_id : int { enum
/external/tensorflow/tensorflow/python/tpu/
Dtensor_tracer.py860 def _skip_op(self, op_id, op, ops_in_exec_path, report_handler): argument
911 def _skip_tensor(self, op_id, out_tensor, report_handler): argument
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_runtime.cc390 xla::int32 channel_id_present, xla::int64 op_id, xla::int32 reduction_kind, in __xla_cpu_runtime_AllReduce()
/external/tensorflow/tensorflow/core/kernels/hexagon/
Dhexagon_control_wrapper.cc272 const int op_id = params.soc_op_id(); in SetupGraph() local

12