Searched refs:OpSegment (Results 1 – 6 of 6) sorted by relevance
28 OpSegment::Item::~Item() { in ~Item()32 OpSegment::OpSegment() {} in OpSegment() function in tensorflow::OpSegment34 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()
42 class OpSegment {44 OpSegment();45 ~OpSegment();84 TF_DISALLOW_COPY_AND_ASSIGN(OpSegment);
64 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()
156 OpSegment* op_segment() { return &op_seg_; } in op_segment()195 OpSegment op_seg_;
255 if (!OpSegment::ShouldOwnKernel(lib, props->node_def.op())) { in InitItem()268 if (kernel && !OpSegment::ShouldOwnKernel(lib, kernel->type_string())) { in InitItem()
1383 if (!OpSegment::ShouldOwnKernel(lib, props->node_def.op())) { in CreateExecutors()1396 if (kernel && !OpSegment::ShouldOwnKernel(lib, kernel->type_string())) in CreateExecutors()