Home
last modified time | relevance | path

Searched defs:lib_def (Results 1 – 25 of 28) sorted by relevance

12

/external/tensorflow/tensorflow/core/distributed_runtime/
Dcluster_function_library_runtime_test.cc57 const FunctionLibraryDefinition& lib_def, GraphDef* g, in ConstructFunctionGraphHelper()
64 const FunctionLibraryDefinition& lib_def, in Instantiate()
74 const string& function_name, const FunctionLibraryDefinition& lib_def, in InstantiateAndRun()
123 FunctionLibraryDefinition lib_def(OpRegistry::Global(), proto); in TEST_F() local
382 FunctionLibraryDefinition lib_def(OpRegistry::Global(), proto); in TEST_F() local
397 FunctionLibraryDefinition lib_def(OpRegistry::Global(), proto); in TEST_F() local
Dcluster_function_library_runtime.cc182 const string& function_name, const FunctionLibraryDefinition& lib_def, in Instantiate()
206 auto construct_graph_fn = [&](const FunctionLibraryDefinition* lib_def) { in Instantiate()
Dgraph_mgr.h132 std::unique_ptr<FunctionLibraryDefinition> lib_def; member
/external/tensorflow/tensorflow/core/common_runtime/
Dfunction_def_utils.cc31 const FunctionLibraryDefinition* const lib_def, in FunctionDefToBodyHelper()
77 const FunctionLibraryDefinition* lib_def, in FunctionDefToBodyHelper()
Dfunction.cc149 const FunctionLibraryDefinition* lib_def) in FunctionLibraryRuntimeOverlay()
410 const FunctionLibraryDefinition* lib_def = nullptr; // Not owned. member
460 int graph_def_version, const FunctionLibraryDefinition* lib_def, in FunctionLibraryRuntimeImpl()
622 const FunctionLibraryDefinition* lib_def = in CreateKernel() local
674 const FunctionLibraryDefinition* lib_def, in FunctionDefToBody()
687 const NameAttrList& func, const FunctionLibraryDefinition* lib_def, in InstantiateSymbolicGradient()
774 const FunctionLibraryDefinition* lib_def = in Instantiate() local
918 const FunctionLibraryDefinition* lib_def = in CreateItem() local
1375 const FunctionLibraryDefinition* lib_def, thread::ThreadPool* thread_pool, in NewFunctionLibraryRuntime()
Dfunction_utils.cc282 bool IsFunctionCall(const FunctionLibraryDefinition& lib_def, in IsFunctionCall()
Dprocess_function_library_runtime.cc65 const FunctionLibraryDefinition& lib_def, AttrSlice attrs, in DistributedInit()
87 int graph_def_version, const FunctionLibraryDefinition* lib_def, in ProcessFunctionLibraryRuntime()
634 const FunctionLibraryDefinition* lib_def, std::unique_ptr<Graph>* graph, in GetGraphAndArgRets()
700 const FunctionLibraryDefinition* lib_def = in InstantiateMultiDevice() local
Dprocess_function_library_runtime_test.cc56 const FunctionLibraryDefinition& lib_def, AttrSlice attrs, in Instantiate()
336 std::unique_ptr<FunctionLibraryDefinition> lib_def( in TEST_F() local
/external/tensorflow/tensorflow/core/framework/
Dfunction_test.cc973 FunctionLibraryDefinition lib_def(OpRegistry::Global(), {}); in TEST() local
981 FunctionLibraryDefinition lib_def(OpRegistry::Global(), {}); in TEST() local
992 FunctionLibraryDefinition lib_def(OpRegistry::Global(), {}); in TEST() local
1011 FunctionLibraryDefinition lib_def(OpRegistry::Global(), {}); in TEST() local
1036 FunctionLibraryDefinition lib_def(OpRegistry::Global(), FunctionDefLibrary()); in TEST() local
1058 FunctionLibraryDefinition lib_def(OpRegistry::Global(), {}); in TEST() local
1072 FunctionLibraryDefinition lib_def(OpRegistry::Global(), {}); in TEST() local
1085 FunctionLibraryDefinition lib_def(OpRegistry::Global(), proto); in TEST() local
1145 FunctionLibraryDefinition lib_def(OpRegistry::Global(), FunctionDefLibrary()); in TEST() local
1184 FunctionLibraryDefinition lib_def(OpRegistry::Global(), proto); in TEST() local
[all …]
Dfunction.h691 const FunctionLibraryDefinition* lib_def = nullptr; member
Ddataset.cc274 const FunctionLibraryDefinition& lib_def) { in AddFunction()
Ddataset.h279 const FunctionLibraryDefinition& lib_def) { in AddAttrFunctions()
/external/tensorflow/tensorflow/core/kernels/data/
Dcaptured_function.h95 const FunctionLibraryDefinition* lib_def() const { return lib_def_.get(); } in lib_def() function
172 const FunctionLibraryDefinition* lib_def() const { in lib_def() function
Dserialization_utils.cc30 FunctionLibraryDefinition lib_def(OpRegistry::Global(), graph_def.library()); in FindStatefulOps() local
Drewrite_utils.cc165 std::unique_ptr<FunctionLibraryDefinition> lib_def = nullptr; in RewriteDataset() local
Dcaptured_function.cc228 const FunctionLibraryDefinition* lib_def, const string& func_name, in CreateFunctionLibraryDefinition()
265 Status LookupFunction(const FunctionLibraryDefinition& lib_def, in LookupFunction()
/external/tensorflow/tensorflow/core/common_runtime/eager/
Dshape_inference.cc29 const FunctionLibraryDefinition& lib_def, in RunShapeInference()
Dkernel_and_device.cc166 const FunctionLibraryDefinition* lib_def = in InstantiateFunc() local
/external/tensorflow/tensorflow/compiler/jit/
Dencapsulate_subgraphs_pass_test.cc484 std::unique_ptr<FunctionLibraryDefinition> lib_def( in Encapsulate() local
616 std::unique_ptr<FunctionLibraryDefinition> lib_def( in TEST() local
676 std::unique_ptr<FunctionLibraryDefinition> lib_def( in TEST() local
950 std::unique_ptr<FunctionLibraryDefinition> lib_def( in TEST() local
1138 std::unique_ptr<FunctionLibraryDefinition> lib_def( in TEST() local
1306 std::unique_ptr<FunctionLibraryDefinition> lib_def( in TEST() local
1469 std::unique_ptr<FunctionLibraryDefinition> lib_def( in TEST() local
1583 std::unique_ptr<FunctionLibraryDefinition> lib_def( in TEST() local
1678 std::unique_ptr<FunctionLibraryDefinition> lib_def( in TEST() local
1786 std::unique_ptr<FunctionLibraryDefinition> lib_def( in TEST() local
[all …]
/external/tensorflow/tensorflow/core/distributed_runtime/eager/
Dremote_execute_node.h44 const NodeDef& ndef, FunctionLibraryDefinition* lib_def, in RemoteExecuteNode()
Dcluster_function_library_runtime.cc49 const string& function_name, const FunctionLibraryDefinition& lib_def, in Instantiate()
/external/tensorflow/tensorflow/core/grappler/optimizers/data/
Dslack_test.cc114 FunctionDefLibrary lib_def; in TEST() local
Dvectorization_utils_test.cc262 FunctionLibraryDefinition lib_def(OpRegistry::Global(), lib); in TEST() local
479 FunctionLibraryDefinition lib_def(OpRegistry::Global(), lib); in TEST() local
Dfunction_utils_test.cc804 FunctionLibraryDefinition lib_def(OpRegistry::Global(), in TEST() local
/external/tensorflow/tensorflow/core/grappler/utils/
Dfunctions_test.cc676 FunctionDefLibrary lib_def; in TEST_F() local

12