| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | const_analysis.cc | 36 Status GetFunctionBody(FunctionLibraryRuntime* flib_runtime, in GetFunctionBody() argument
42 TF_RETURN_IF_ERROR(flib_runtime->Instantiate( in GetFunctionBody()
44 *fbody = flib_runtime->GetFunctionBody(func_handle); in GetFunctionBody()
48 Status GetFunctionBodies(FunctionLibraryRuntime* flib_runtime, in GetFunctionBodies() argument
55 TF_RETURN_IF_ERROR(flib_runtime->Instantiate( in GetFunctionBodies()
58 fbodies->push_back(flib_runtime->GetFunctionBody(func_handle)); in GetFunctionBodies()
65 std::vector<int>* const_input_idxs, FunctionLibraryRuntime* flib_runtime) { in CondConstInputIndices() argument
76 /*compile_time_const_nodes=*/nullptr, flib_runtime)); in CondConstInputIndices()
92 FunctionLibraryRuntime* flib_runtime) { in GetCompileTimeConstInputs() argument
98 TF_RETURN_IF_ERROR(GetFunctionBody(flib_runtime, node, "cond", &fcond)); in GetCompileTimeConstInputs()
[all …]
|
| D | const_analysis.h | 41 FunctionLibraryRuntime* flib_runtime,
48 FunctionLibraryRuntime* flib_runtime);
|
| D | xla_compiler.h | 288 FunctionLibraryRuntime* flib_runtime() const { return flib_runtime_; } in flib_runtime() function
|
| D | xla_compiler.cc | 542 FunctionLibraryRuntime* flib_runtime, in GetFunctionBody() argument
545 TF_RETURN_IF_ERROR(flib_runtime->Instantiate( in GetFunctionBody()
548 *fbody = flib_runtime->GetFunctionBody(handle); in GetFunctionBody()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | compilability_check_util_test.cc | 126 auto* flib_runtime = GetFunctionLibraryRuntime(); in TEST_F() local
128 EXPECT_FALSE(checker_->IsCompilableNode(*const0, flib_runtime)); in TEST_F()
130 EXPECT_TRUE(checker_->IsCompilableNode(*compilable_op, flib_runtime)); in TEST_F()
133 EXPECT_FALSE(checker_->IsCompilableNode(*uncompilable_op, flib_runtime)); in TEST_F()
136 checker_->FindUncompilableNodes(*uncompilable_op, flib_runtime); in TEST_F()
159 auto* flib_runtime = GetFunctionLibraryRuntime(); in TEST_F() local
162 EXPECT_FALSE(checker_->IsCompilableNode(*uncompilable_op, flib_runtime)); in TEST_F()
165 checker_->FindUncompilableNodes(*uncompilable_op, flib_runtime); in TEST_F()
172 EXPECT_TRUE(checker_->IsCompilableNode(*uncompilable_op, flib_runtime)); in TEST_F()
175 checker_->FindUncompilableNodes(*uncompilable_op, flib_runtime); in TEST_F()
[all …]
|
| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| D | tpu_compile_op_common.cc | 145 FunctionLibraryRuntime* flib_runtime, in CompileLocallyAndFillHostCache() argument
157 const ConfigProto* config = flib_runtime->config_proto(); in CompileLocallyAndFillHostCache()
166 flib_runtime->GetFunctionLibraryDefinition(), in CompileLocallyAndFillHostCache()
167 flib_runtime->graph_def_version(), in CompileLocallyAndFillHostCache()
|
| D | tpu_compile_op_common.h | 117 FunctionLibraryRuntime* flib_runtime,
128 FunctionLibraryRuntime* flib_runtime, in LookupPersistentCompilationCacheAndFillCaches() argument
|
| /external/tensorflow/tensorflow/core/tpu/ |
| D | tpu_compile.cc | 402 TF_RETURN_IF_ERROR(compiler->flib_runtime()->Instantiate( in CompileTFFunctionToHlo()
404 const FunctionBody* fbody = compiler->flib_runtime()->GetFunctionBody(handle); in CompileTFFunctionToHlo()
437 compiler->flib_runtime(), &flib_definition)); in CompileTFFunctionToHlo()
|
| /external/tensorflow/tensorflow/core/tpu/kernels/xla/ |
| D | host_compute_ops.cc | 114 FunctionLibraryRuntime* flib_runtime = ctx->function_library(); in HostComputeOp() local
115 OP_REQUIRES(ctx, flib_runtime != nullptr, in HostComputeOp()
119 flib_runtime->GetFunctionLibraryDefinition(); in HostComputeOp()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/kernels/ |
| D | trt_engine_op.cc | 286 FunctionLibraryRuntime* flib_runtime, in FunctionDefToGraphDef() argument
289 flib_runtime->GetFunctionLibraryDefinition(); in FunctionDefToGraphDef()
291 fbody = flib_runtime->GetFunctionBody(handle); in FunctionDefToGraphDef()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | while_op.cc | 124 ctx->compiler()->flib_runtime()->GetFunctionLibraryDefinition(); in GetLoopInvariants()
|