Searched refs:flib_ (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | graph_compiler.cc | 133 Status s = flib_->CreateKernel(n->def(), &op_kernel_raw); in Compile()
166 if (IsFunctionCall(*flib_->GetFunctionLibraryDefinition(), *n)) { in Compile()
221 TF_RET_CHECK(IsFunctionCall(*flib_->GetFunctionLibraryDefinition(), *n)); in CompileFunctionalNode()
232 TF_RETURN_IF_ERROR(GetFunctionNameAndAttr(*flib_, *n, &func)); in CompileFunctionalNode()
311 params->function_library = flib_; in PartiallySetupParams()
|
| D | graph_compiler.h | 63 flib_(flib), in GraphCompiler()
83 FunctionLibraryRuntime* flib_; variable
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | lower_if_op.cc | 94 const FunctionLibraryDefinition& flib_; member in tensorflow::__anon2655e6fa0111::CondBuilder
107 flib_(flib), in CondBuilder()
286 TF_RETURN_IF_ERROR(InlineCallInGraph("then", then_call_node_, flib_, graph_)); in InlineCallNodes()
287 TF_RETURN_IF_ERROR(InlineCallInGraph("else", else_call_node_, flib_, graph_)); in InlineCallNodes()
|
| D | lower_while_op.cc | 133 const FunctionLibraryDefinition& flib_; member in tensorflow::__anonc08876440111::LowerWhileHelper
158 flib_(flib), in LowerWhileHelper()
407 TF_RETURN_IF_ERROR(InlineCallInGraph("cond", cond_call_node_, graph_, flib_)); in InlineCallNodes()
408 TF_RETURN_IF_ERROR(InlineCallInGraph("body", body_call_node_, graph_, flib_)); in InlineCallNodes()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | op_kernel.h | 353 FunctionLibraryRuntime* function_library() const { return flib_; } in function_library()
382 FunctionLibraryRuntime* flib_; variable
|
| D | op_kernel.cc | 213 flib_(flib), in OpKernelConstruction()
|