Home
last modified time | relevance | path

Searched refs:OpSegment (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/core/framework/
Dop_segment.cc28 OpSegment::Item::~Item() { in ~Item()
32 OpSegment::OpSegment() {} in OpSegment() function in tensorflow::OpSegment
34 OpSegment::~OpSegment() { in ~OpSegment()
38 Status OpSegment::FindOrCreate(const string& session_handle, in FindOrCreate()
74 void OpSegment::AddHold(const string& session_handle) { in AddHold()
84 void OpSegment::RemoveHold(const string& session_handle) { in RemoveHold()
103 bool OpSegment::ShouldOwnKernel(FunctionLibraryRuntime* lib, in ShouldOwnKernel()
Dop_segment.h42 class OpSegment {
44 OpSegment();
45 ~OpSegment();
83 TF_DISALLOW_COPY_AND_ASSIGN(OpSegment);
Dop_segment_test.cc64 OpSegment::CreateKernelFn GetFn(const NodeDef* ndef) { in GetFn()
78 OpSegment opseg; in TEST_F()
115 OpSegment opseg; in TEST_F()
123 OpSegment opseg; in TEST_F()
134 OpSegment opseg; in TEST_F()
/external/tensorflow/tensorflow/core/common_runtime/
Ddevice.h172 OpSegment* op_segment() { return &op_seg_; } in op_segment()
215 OpSegment op_seg_;
Ddirect_session.cc1269 if (!OpSegment::ShouldOwnKernel(lib, ndef.op())) { in CreateExecutors()
1282 if (kernel && !OpSegment::ShouldOwnKernel(lib, kernel->type_string())) in CreateExecutors()
/external/tensorflow/tensorflow/core/distributed_runtime/
Dgraph_mgr.cc242 if (!OpSegment::ShouldOwnKernel(lib, ndef.op())) { in InitItem()
254 if (kernel && !OpSegment::ShouldOwnKernel(lib, kernel->type_string())) { in InitItem()