| /external/tensorflow/tensorflow/compiler/jit/ |
| D | build_xla_ops_pass_test.cc | 58 Status BuildXlaOps(const Scope& s, const FunctionDefLibrary& fdef_lib, in BuildXlaOps() argument
62 FunctionLibraryDefinition flib_def(graph->op_registry(), fdef_lib); in BuildXlaOps()
123 FunctionDefLibrary fdef_lib; in CreateFunctionDefLibWithConstFunction() local
129 *fdef_lib.add_function() = std::move(func); in CreateFunctionDefLibWithConstFunction()
130 return fdef_lib; in CreateFunctionDefLibWithConstFunction()
137 FunctionDefLibrary fdef_lib = in TEST_F() local
139 TF_ASSERT_OK(root.graph()->AddFunctionLibrary(fdef_lib)); in TEST_F()
149 TF_ASSERT_OK(BuildXlaOps(root, fdef_lib, &graph)); in TEST_F()
159 FunctionDefLibrary fdef_lib = in TEST_F() local
161 TF_ASSERT_OK(root.graph()->AddFunctionLibrary(fdef_lib)); in TEST_F()
[all …]
|
| D | xla_cluster_util_test.cc | 206 void AddRefFunctionFunctionDef(FunctionDefLibrary* fdef_lib) { in AddRefFunctionFunctionDef() argument
214 *fdef_lib->add_function() = make_ref_float; in AddRefFunctionFunctionDef()
217 void AddRegularFunctionFunctionDef(FunctionDefLibrary* fdef_lib) { in AddRegularFunctionFunctionDef() argument
223 *fdef_lib->add_function() = make_regular_float; in AddRegularFunctionFunctionDef()
229 FunctionDefLibrary fdef_lib; in TEST() local
234 AddRefFunctionFunctionDef(&fdef_lib); in TEST()
237 AddRegularFunctionFunctionDef(&fdef_lib); in TEST()
240 FunctionLibraryDefinition flib_def(OpRegistry::Global(), fdef_lib); in TEST()
|
| /external/tensorflow/tensorflow/c/ |
| D | c_api_function.cc | 253 tensorflow::FunctionDefLibrary fdef_lib; in TF_GraphCopyFunction() local
254 *fdef_lib.add_function() = func->fdef; in TF_GraphCopyFunction()
256 *fdef_lib.add_function() = grad->fdef; in TF_GraphCopyFunction()
257 tensorflow::GradientDef* gdef = fdef_lib.add_gradient(); in TF_GraphCopyFunction()
263 status->status = g->graph.AddFunctionLibrary(fdef_lib); in TF_GraphCopyFunction()
|
| D | c_api_experimental.cc | 205 const auto& fdef_lib = gdef.library(); in CreateFunctionsFromTextProto() local
206 if (fdef_lib.gradient_size() > 0) { in CreateFunctionsFromTextProto()
213 for (const FunctionDef& fdef : fdef_lib.function()) { in CreateFunctionsFromTextProto()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | replicate_per_replica_nodes_test.cc | 229 FunctionDefLibrary fdef_lib; in TEST() local
230 FunctionLibraryDefinition flib_def(OpRegistry::Global(), fdef_lib); in TEST()
244 *fdef_lib.add_function() = fdef; in TEST()
250 TF_EXPECT_OK(scope.graph()->AddFunctionLibrary(fdef_lib)); in TEST()
|
| /external/tensorflow/tensorflow/core/graph/ |
| D | graph_def_builder.h | 168 Status AddFunctionLibrary(const FunctionDefLibrary& fdef_lib) { in AddFunctionLibrary() argument
169 return graph_.AddFunctionLibrary(fdef_lib); in AddFunctionLibrary()
|
| D | graph_partition_test.cc | 462 FunctionDefLibrary fdef_lib; in TEST_F() local
463 *fdef_lib.add_function() = test::function::XTimesTwo(); in TEST_F()
464 *fdef_lib.add_function() = test::function::XTimesFour(); in TEST_F()
465 TF_ASSERT_OK(in_.graph()->AddFunctionLibrary(fdef_lib)); in TEST_F()
|
| D | graph.cc | 656 Status Graph::AddFunctionLibrary(const FunctionDefLibrary& fdef_lib) { in AddFunctionLibrary() argument
658 if (fdef_lib.function_size() > 0 && versions_->min_consumer() < 12) { in AddFunctionLibrary()
661 return ops_.AddLibrary(fdef_lib); in AddFunctionLibrary()
|
| D | graph.h | 583 Status AddFunctionLibrary(const FunctionDefLibrary& fdef_lib);
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | functionalize_control_flow.cc | 125 FunctionDefLibrary fdef_lib; in AddFunctionDefToGraphLibrary() local
126 *(fdef_lib.add_function()) = *new_fdef; in AddFunctionDefToGraphLibrary()
127 return graph->AddFunctionLibrary(fdef_lib); in AddFunctionDefToGraphLibrary()
|
| D | functionalize_cond_test.cc | 144 FunctionDefLibrary fdef_lib; in TEST() local
145 FunctionLibraryDefinition flib_def(OpRegistry::Global(), fdef_lib); in TEST()
|
| D | functionalize_control_flow_test.cc | 226 FunctionDefLibrary fdef_lib; in RunTest() local
227 *(fdef_lib.add_function()) = cond_fdef; in RunTest()
228 TF_ASSERT_OK(scope.graph()->AddFunctionLibrary(fdef_lib)); in RunTest()
448 FunctionDefLibrary fdef_lib; in AddNoinlineFunctionToGraph() local
449 *(fdef_lib.add_function()) = GetNoinlineFunctionDef(); in AddNoinlineFunctionToGraph()
450 TF_RETURN_IF_ERROR(graph->AddFunctionLibrary(fdef_lib)); in AddNoinlineFunctionToGraph()
|
| D | xla_compiler_test.cc | 641 FunctionDefLibrary fdef_lib; in TEST_F() local
642 *(fdef_lib.add_function()) = fdef; in TEST_F()
646 TF_EXPECT_OK(scope.graph()->AddFunctionLibrary(fdef_lib)); in TEST_F()
681 FunctionLibraryDefinition flib_def(OpRegistry::Global(), fdef_lib); in TEST_F()
1570 FunctionDefLibrary fdef_lib; in TEST_F() local
1571 FunctionLibraryDefinition flib_def(OpRegistry::Global(), fdef_lib); in TEST_F()
1642 FunctionDefLibrary fdef_lib; in TEST_F() local
1643 FunctionLibraryDefinition flib_def(OpRegistry::Global(), fdef_lib); in TEST_F()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | meta_optimizer.cc | 101 const FunctionDefLibrary& fdef_lib) { in GetFunctionDefLibraryStub() argument
103 for (const FunctionDef& fn : fdef_lib.function()) { in GetFunctionDefLibraryStub()
110 *stub.mutable_gradient() = fdef_lib.gradient(); in GetFunctionDefLibraryStub()
623 FunctionDefLibrary& fdef_lib) { in PropagateTFDataAttrs() argument
660 for (const auto& func_def : fdef_lib.function()) { in PropagateTFDataAttrs()
668 for (FunctionDef& func_def : *fdef_lib.mutable_function()) { in PropagateTFDataAttrs()
|
| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| D | tpu_compile.proto | 55 FunctionDefLibrary fdef_lib = 9; field
|