Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/tf2tensorrt/kernels/
Dtrt_engine_op.cc110 string funcdef_name_; member in tensorflow::tensorrt::TRTEngineOp
167 auto fdef = lib->GetFunctionLibraryDefinition()->Find(funcdef_name_); in ConstructFunctionHandle()
169 return errors::Internal("Native FunctionDef ", funcdef_name_, in ConstructFunctionHandle()
177 auto status = lib->Instantiate(funcdef_name_, AttrSlice(&fdef->attr()), in ConstructFunctionHandle()
180 LOG(ERROR) << " Instantiating native function " << funcdef_name_ in ConstructFunctionHandle()
215 context->GetAttr("segment_funcdef_name", &funcdef_name_)); in TRTEngineOp()
244 if (funcdef_name_.empty()) { in ExecuteNativeSegment()
255 LOG(ERROR) << "Couldn't construct function handle " << funcdef_name_; in ExecuteNativeSegment()