Home
last modified time | relevance | path

Searched refs:flib_def (Results 1 – 25 of 74) sorted by relevance

123

/external/tensorflow/tensorflow/core/kernels/data/
Diterator_ops.cc60 std::unique_ptr<FunctionLibraryDefinition> flib_def, in IteratorResource() argument
66 std::move(flib_def), std::move(pflr), lib, nullptr /* iterator */)), in IteratorResource()
112 params.flib_def = captured_state->flib_def.get(); in Save()
148 std::unique_ptr<FunctionLibraryDefinition> flib_def(nullptr); in Restore() local
150 TF_RETURN_IF_ERROR(ctx->function_library()->Clone(&flib_def, &pflr, &lib)); in Restore()
158 AddToFunctionLibrary(flib_def.get(), graph_def.library())); in Restore()
160 std::move(flib_def), std::move(pflr), lib, nullptr /* iterator */); in Restore()
198 Status AddLibrary(const FunctionLibraryDefinition& flib_def) { in AddLibrary() argument
200 return iterator_state_->flib_def->AddLibrary(flib_def); in AddLibrary()
208 iterator_state_->flib_def, iterator_state_->pflr, in SetIteratorFromDataset()
[all …]
Dmulti_device_iterator_ops.cc49 std::unique_ptr<FunctionLibraryDefinition> flib_def, in MultiDeviceIterator() argument
57 flib_def_(std::move(flib_def)), in MultiDeviceIterator()
406 std::unique_ptr<FunctionLibraryDefinition> flib_def(nullptr); in Compute() local
409 &flib_def, &pflr, &lib)); in Compute()
423 std::move(flib_def), std::move(pflr), lib, in Compute()
435 [this, context, lib, &flib_def, &pflr, in Compute()
441 std::move(flib_def), std::move(pflr), in Compute()
/external/tensorflow/tensorflow/compiler/jit/
Dmark_for_compilation_pass_test_helper.cc23 std::unique_ptr<Graph>* graph, FunctionLibraryDefinition* flib_def, in MarkForCompilation() argument
50 opt_options.flib_def = flib_def; in MarkForCompilation()
58 FunctionLibraryDefinition flib_def((*graph)->op_registry(), flib); in MarkForCompilation() local
59 return MarkForCompilation(graph, &flib_def, enable_global_jit); in MarkForCompilation()
Dencapsulate_xla_computations_pass_test.cc36 const FunctionLibraryDefinition& flib_def, const string& function) { in MakeOuterGraph() argument
38 TF_EXPECT_OK(scope.graph()->AddFunctionLibrary(flib_def.ToProto())); in MakeOuterGraph()
50 NodeDefBuilder("launch0", function, &flib_def) in MakeOuterGraph()
147 FunctionLibraryDefinition flib_def(OpRegistry::Global(), {}); in TEST() local
148 std::unique_ptr<Graph> graph(new Graph(&flib_def)); in TEST()
181 TF_CHECK_OK(EncapsulateXlaComputationsPass::Encapsulate(&graph, &flib_def)); in TEST()
206 FunctionLibraryDefinition flib_def(OpRegistry::Global(), {}); in TEST() local
207 std::unique_ptr<Graph> graph(new Graph(&flib_def)); in TEST()
254 std::unique_ptr<Graph> graph_copy(new Graph(&flib_def)); in TEST()
257 TF_ASSERT_OK(EncapsulateXlaComputationsPass::Encapsulate(&graph, &flib_def)); in TEST()
[all …]
Dresource_operation_safety_analysis_test.cc76 return ComputeIncompatibleResourceOperationPairs(*g, &g->flib_def(), {}, in ComputeIncompatiblePairs()
243 FunctionDefLibrary flib_def; in CreateFunctionDefLibWithConstFunction() local
249 *flib_def.add_function() = std::move(func); in CreateFunctionDefLibWithConstFunction()
250 return flib_def; in CreateFunctionDefLibWithConstFunction()
264 FunctionDefLibrary flib_def = in TEST() local
266 TF_ASSERT_OK(root.graph()->AddFunctionLibrary(flib_def)); in TEST()
286 FunctionDefLibrary flib_def = in TEST() local
288 TF_ASSERT_OK(root.graph()->AddFunctionLibrary(flib_def)); in TEST()
306 FunctionDefLibrary flib_def = in TEST() local
308 TF_ASSERT_OK(root.graph()->AddFunctionLibrary(flib_def)); in TEST()
[all …]
Dresource_operation_safety_analysis.cc97 Status MayCallFunction(const Node& n, const FunctionLibraryDefinition* flib_def, in MayCallFunction() argument
99 if (flib_def->Contains(n.type_string())) { in MayCallFunction()
116 const Node& n, const FunctionLibraryDefinition* flib_def, in XlaResourceOpKindForNode() argument
138 TF_RETURN_IF_ERROR(MayCallFunction(n, flib_def, &may_call_function)); in XlaResourceOpKindForNode()
267 const Graph& g, const FunctionLibraryDefinition* flib_def, in ComputeIncompatibleResourceOperationPairs() argument
286 *n, flib_def, resource_ops_to_ignore, &op_kind)); in ComputeIncompatibleResourceOperationPairs()
Dbuild_xla_ops_pass.cc234 const FunctionLibraryDefinition& flib_def, Node* n, Graph* g, in ReplaceFunctionCallWithPartionedCall() argument
279 const FunctionLibraryDefinition& flib_def, in InferDeviceForCluster() argument
281 const FunctionDef* func_def = flib_def.Find(function_name); in InferDeviceForCluster()
312 const FunctionLibraryDefinition& flib_def, bool lazy_compilation_enabled, in ReplaceNodeWithXlaCompileAndXlaRun() argument
319 flib_def, &device)); in ReplaceNodeWithXlaCompileAndXlaRun()
394 options, flib_def, n, g, cluster_info.function, root)); in ReplaceNodeWithXlaCompileAndXlaRun()
425 options, *options.flib_def, lazy_compilation_enabled, graph, n)); in Run()
429 DumpGraphToFile("build_xla_ops", *graph, options.flib_def); in Run()
Dencapsulate_xla_computations_pass.cc211 std::unique_ptr<Graph>* graph, FunctionLibraryDefinition* flib_def) { in Encapsulate() argument
236 /*reuse_existing_functions=*/true, &output, flib_def), in Encapsulate()
376 **options.graph, options.flib_def); in Run()
383 TF_RETURN_WITH_CONTEXT_IF_ERROR(Encapsulate(options.graph, options.flib_def), in Run()
387 **options.graph, options.flib_def); in Run()
393 **options.graph, options.flib_def); in Run()
Dpartially_decluster_pass.cc282 const FunctionLibraryDefinition* flib_def, in PartiallyDeclusterGraph() argument
287 nullptr, env, TF_GRAPH_DEF_VERSION, flib_def, opts); in PartiallyDeclusterGraph()
356 if (options.flib_def == nullptr) { in Run()
368 graph, options.flib_def, options.session_options->env)); in Run()
Dmark_for_compilation_pass.cc449 const Graph& graph, FunctionLibraryDefinition* flib_def, Env* env, in FindCompilationCandidates() argument
455 flib_def, opts)); in FindCompilationCandidates()
683 const FunctionLibraryDefinition* fld = options.flib_def; in Run()
1089 *graph, options.flib_def, in RunImpl()
1106 graph, options.flib_def, IgnoreResourceOpForSafetyAnalysis, &cycles)); in RunImpl()
1127 if (options.flib_def->GetAttr(*node, kXlaCompileAttr, &xla_compile_attr) in RunImpl()
1291 options.flib_def); in RunImpl()
1321 } else if (options.flib_def->GetAttr(*n, kXlaCompileAttr, &compile_attr) in RunImpl()
1345 DumpGraphToFile("mark_for_compilation", **options.graph, options.flib_def); in RunImpl()
1368 options.flib_def); in RunImpl()
Dbuild_xla_ops_pass_test.cc61 FunctionLibraryDefinition flib_def(graph->op_registry(), fdef_lib); in BuildXlaOps() local
78 opt_options.flib_def = &flib_def; in BuildXlaOps()
Dresource_operation_safety_analysis.h68 const Graph& g, const FunctionLibraryDefinition* flib_def,
/external/tensorflow/tensorflow/core/common_runtime/data/
Dstandalone.cc55 std::unique_ptr<FunctionLibraryDefinition> flib_def = in FromGraph() local
56 MakeUnique<FunctionLibraryDefinition>(graph.flib_def()); in FromGraph()
60 flib_def.get(), OptimizerOptions{}, nullptr /* parent */); in FromGraph()
80 flib_def.release(), pool.release())); in FromGraph()
113 FunctionLibraryDefinition* flib_def, thread::ThreadPool* pool) in Dataset() argument
116 flib_def_(flib_def), in Dataset()
/external/tensorflow/tensorflow/core/common_runtime/
Dlower_if_op_test.cc47 FunctionLibraryDefinition flib_def((*graph)->flib_def()); in Rewrite() local
50 opt_options.flib_def = &flib_def; in Rewrite()
71 TF_ASSERT_OK(NodeBuilder("if", "If", &root.graph()->flib_def()) in TEST()
189 TF_ASSERT_OK(NodeBuilder("if", "If", &root.graph()->flib_def()) in TEST()
277 TF_ASSERT_OK(NodeBuilder("if", "If", &root.graph()->flib_def()) in TEST()
Dlower_while_op_test.cc39 FunctionLibraryDefinition flib_def((*graph)->flib_def()); in Rewrite() local
42 opt_options.flib_def = &flib_def; in Rewrite()
64 TF_ASSERT_OK(NodeBuilder("while", "While", &root.graph()->flib_def()) in TEST()
167 TF_ASSERT_OK(NodeBuilder("while", "While", &root.graph()->flib_def()) in TEST()
269 TF_ASSERT_OK(NodeBuilder("while", "While", &root.graph()->flib_def()) in TEST()
Dlower_if_while_test.cc47 FunctionLibraryDefinition flib_def((*graph)->flib_def()); in Rewrite() local
50 opt_options.flib_def = &flib_def; in Rewrite()
150 TF_ASSERT_OK(NodeBuilder("while", "While", &root.graph()->flib_def()) in TEST()
255 TF_ASSERT_OK(NodeBuilder("if", "If", &root.graph()->flib_def()) in TEST()
Dgraph_execution_state.h57 : flib_def(std::move(flib)), in ClientGraph()
58 graph(flib_def.get()), in ClientGraph()
64 std::unique_ptr<FunctionLibraryDefinition> flib_def; member
Dpartitioning_utils.h74 FunctionNameGenerator(const FunctionLibraryDefinition* flib_def, in FunctionNameGenerator() argument
76 : flib_def_(flib_def), name_(name), counter_(0) {} in FunctionNameGenerator()
Doptimization_registry_test.cc49 return options.flib_def->AddFunctionDef(test::function::WXPlusB()); in Run()
67 options.flib_def = flib_def_.get(); in RunPass()
Doptimization_registry.cc49 **options.graph, options.flib_def); in RunGrouping()
58 *part.second, options.flib_def); in RunGrouping()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dfunctionalize_control_flow.cc236 options.flib_def); in Run()
241 TF_GRAPH_DEF_VERSION, options.flib_def, OptimizerOptions())); in Run()
266 string new_func_name = options.flib_def->UniqueFunctionName( in Run()
270 func.name(), new_func_name, func.attr(), options.flib_def, flr, in Run()
281 options.flib_def); in Run()
Dtf2xla.cc293 compiler_options.flib_def = &graph->flib_def(); in ConvertGraphToXla()
344 FunctionLibraryDefinition flib_def(OpRegistry::Global(), graph_def.library()); in InitGraph() local
345 std::unique_ptr<Graph> g(new Graph(flib_def)); in InitGraph()
370 TF_RETURN_IF_ERROR(FunctionalizeControlFlow(g.get(), &flib_def)); in InitGraph()
373 TF_RETURN_IF_ERROR(g->AddFunctionLibrary(flib_def.ToProto())); in InitGraph()
/external/tensorflow/tensorflow/core/util/
Ddump_graph.cc119 const FunctionLibraryDefinition* flib_def, in DumpGraphToFile() argument
123 if (flib_def) { in DumpGraphToFile()
124 *graph_def.mutable_library() = flib_def->ToProto(); in DumpGraphToFile()
/external/tensorflow/tensorflow/tools/graph_transforms/
Dadd_default_attributes.cc28 std::unique_ptr<FunctionLibraryDefinition> flib_def( in AddDefaultAttributes() local
33 TF_RETURN_IF_ERROR(AddDefaultAttrsToGraphDef(output_graph_def, *flib_def, 0)); in AddDefaultAttributes()
/external/tensorflow/tensorflow/core/framework/
Ddataset.cc258 EnsureFunctionIsStateless(ctx->flib_def(), function_name)); in AddFunction()
260 const FunctionDef* f_def = ctx->flib_def().Find(function_name); in AddFunction()
267 const string gradient_func = ctx->flib_def().FindGradient(function_name); in AddFunction()
278 TF_RETURN_IF_ERROR(ctx->flib_def().LookUp(node_def.op(), &op_reg_data)); in AddFunction()

123