Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/tf2xla/
Dconst_analysis.cc35 Status GetFunctionBody(FunctionLibraryRuntime* flib_runtime, in GetFunctionBody() argument
41 TF_RETURN_IF_ERROR(flib_runtime->Instantiate( in GetFunctionBody()
43 *fbody = flib_runtime->GetFunctionBody(func_handle); in GetFunctionBody()
47 Status GetFunctionBodies(FunctionLibraryRuntime* flib_runtime, in GetFunctionBodies() argument
54 TF_RETURN_IF_ERROR(flib_runtime->Instantiate( in GetFunctionBodies()
57 fbodies->push_back(flib_runtime->GetFunctionBody(func_handle)); in GetFunctionBodies()
64 std::vector<int>* const_input_idxs, FunctionLibraryRuntime* flib_runtime) { in CondConstInputIndices() argument
75 /*compile_time_const_nodes=*/nullptr, flib_runtime)); in CondConstInputIndices()
91 FunctionLibraryRuntime* flib_runtime) { in GetCompileTimeConstInputs() argument
97 TF_RETURN_IF_ERROR(GetFunctionBody(flib_runtime, node, "cond", &fcond)); in GetCompileTimeConstInputs()
[all …]
Dconst_analysis.h41 FunctionLibraryRuntime* flib_runtime,
48 FunctionLibraryRuntime* flib_runtime);
Dxla_compiler.h285 FunctionLibraryRuntime* flib_runtime() const { return flib_runtime_; } in flib_runtime() function
Dxla_compiler.cc541 FunctionLibraryRuntime* flib_runtime, in GetFunctionBody() argument
544 TF_RETURN_IF_ERROR(flib_runtime->Instantiate( in GetFunctionBody()
547 *fbody = flib_runtime->GetFunctionBody(handle); in GetFunctionBody()
/external/tensorflow/tensorflow/compiler/jit/
Dcompilability_check_util_test.cc125 auto* flib_runtime = GetFunctionLibraryRuntime(); in TEST_F() local
127 EXPECT_FALSE(checker_->IsCompilableNode(*const0, flib_runtime)); in TEST_F()
129 EXPECT_TRUE(checker_->IsCompilableNode(*compilable_op, flib_runtime)); in TEST_F()
132 EXPECT_FALSE(checker_->IsCompilableNode(*uncompilable_op, flib_runtime)); in TEST_F()
135 checker_->FindUncompilableNodes(*uncompilable_op, flib_runtime); in TEST_F()
167 auto* flib_runtime = GetFunctionLibraryRuntime(); in TEST_F() local
168 EXPECT_FALSE(checker_->IsCompilableNode(*functional_node, flib_runtime)); in TEST_F()
170 checker_->FindUncompilableNodes(*functional_node, flib_runtime); in TEST_F()
238 auto* flib_runtime = GetFunctionLibraryRuntime(); in TEST_F() local
240 EXPECT_FALSE(checker_->IsCompilableNode(**while_node_it, flib_runtime)); in TEST_F()
[all …]
/external/tensorflow/tensorflow/core/tpu/kernels/
Dtpu_compile_op_common.h119 FunctionLibraryRuntime* flib_runtime,
130 FunctionLibraryRuntime* flib_runtime, in LookupPersistentCompilationCacheAndFillCaches() argument
Dtpu_compile_op_common.cc312 TF_RETURN_IF_ERROR(compiler->flib_runtime()->Instantiate( in CompileTFFunctionToHlo()
314 const FunctionBody* fbody = compiler->flib_runtime()->GetFunctionBody(handle); in CompileTFFunctionToHlo()
377 compiler->flib_runtime(), &flib_definition)); in CompileTFFunctionToHlo()
591 FunctionLibraryRuntime* flib_runtime, in CompileLocallyAndFillHostCache() argument
608 flib_runtime->GetFunctionLibraryDefinition(), in CompileLocallyAndFillHostCache()
609 flib_runtime->graph_def_version(), in CompileLocallyAndFillHostCache()
/external/tensorflow/tensorflow/core/tpu/kernels/xla/
Dhost_compute_ops.cc102 FunctionLibraryRuntime* flib_runtime = ctx->function_library(); in HostComputeOp() local
103 OP_REQUIRES(ctx, flib_runtime != nullptr, in HostComputeOp()
107 flib_runtime->GetFunctionLibraryDefinition(); in HostComputeOp()
/external/tensorflow/tensorflow/compiler/tf2tensorrt/kernels/
Dtrt_engine_op.cc233 FunctionLibraryRuntime* flib_runtime, in FunctionDefToGraphDef() argument
236 flib_runtime->GetFunctionLibraryDefinition(); in FunctionDefToGraphDef()
238 fbody = flib_runtime->GetFunctionBody(handle); in FunctionDefToGraphDef()