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.cc181 const string& function_name, const FunctionLibraryDefinition& lib_def, in Instantiate()
205 auto construct_graph_fn = [&](const FunctionLibraryDefinition* lib_def) { in Instantiate()
Dgraph_mgr.h135 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.cc153 const FunctionLibraryDefinition* lib_def) in FunctionLibraryRuntimeOverlay()
414 const FunctionLibraryDefinition* lib_def = nullptr; // Not owned. member
468 int graph_def_version, const FunctionLibraryDefinition* lib_def, in FunctionLibraryRuntimeImpl()
627 const FunctionLibraryDefinition* lib_def = in CreateKernel() local
679 const FunctionLibraryDefinition* lib_def, in FunctionDefToBody()
692 const NameAttrList& func, const FunctionLibraryDefinition* lib_def, in InstantiateSymbolicGradient()
786 const FunctionLibraryDefinition* lib_def = in Instantiate() local
936 const FunctionLibraryDefinition* lib_def = in CreateItem() local
1412 const FunctionLibraryDefinition* lib_def, thread::ThreadPool* thread_pool, in NewFunctionLibraryRuntime()
Dprocess_function_library_runtime.cc68 const FunctionLibraryDefinition& lib_def, AttrSlice attrs, in DistributedInit()
90 int graph_def_version, const FunctionLibraryDefinition* lib_def, in ProcessFunctionLibraryRuntime()
462 const FunctionLibraryDefinition* lib_def, Device* default_device) { in PinArgsAndRets()
721 const FunctionLibraryDefinition* lib_def, std::unique_ptr<Graph>* graph, in GetGraphAndArgRets()
787 const FunctionLibraryDefinition* lib_def = in InstantiateMultiDevice() local
Dfunction_utils.cc282 bool IsFunctionCall(const FunctionLibraryDefinition& lib_def, in IsFunctionCall()
Dprocess_function_library_runtime_test.cc58 const FunctionLibraryDefinition& lib_def, AttrSlice attrs, in Instantiate()
330 std::unique_ptr<FunctionLibraryDefinition> lib_def( in TEST_F() local
344 std::unique_ptr<FunctionLibraryDefinition> lib_def( in TEST_F() local
/external/tensorflow/tensorflow/core/framework/
Dfunction_test.cc1040 FunctionLibraryDefinition lib_def(OpRegistry::Global(), {}); in TEST() local
1048 FunctionLibraryDefinition lib_def(OpRegistry::Global(), {}); in TEST() local
1059 FunctionLibraryDefinition lib_def(OpRegistry::Global(), {}); in TEST() local
1078 FunctionLibraryDefinition lib_def(OpRegistry::Global(), {}); in TEST() local
1103 FunctionLibraryDefinition lib_def(OpRegistry::Global(), FunctionDefLibrary()); in TEST() local
1125 FunctionLibraryDefinition lib_def(OpRegistry::Global(), {}); in TEST() local
1139 FunctionLibraryDefinition lib_def(OpRegistry::Global(), {}); in TEST() local
1152 FunctionLibraryDefinition lib_def(OpRegistry::Global(), proto); in TEST() local
1212 FunctionLibraryDefinition lib_def(OpRegistry::Global(), FunctionDefLibrary()); in TEST() local
1251 FunctionLibraryDefinition lib_def(OpRegistry::Global(), proto); in TEST() local
[all …]
Dfunction.h713 const FunctionLibraryDefinition* lib_def = nullptr; member
Ddataset.h323 const FunctionLibraryDefinition& lib_def) { in AddAttrFunctions()
Ddataset.cc325 const FunctionLibraryDefinition& lib_def) { in AddFunction()
/external/tensorflow/tensorflow/core/data/
Dcaptured_function.h96 const FunctionLibraryDefinition* lib_def() const { return lib_def_.get(); } in lib_def() function
197 const FunctionLibraryDefinition* lib_def() const { in lib_def() function
Dserialization_utils.cc62 std::unique_ptr<FunctionLibraryDefinition> lib_def = nullptr; in FromGraphDef() local
79 FunctionLibraryDefinition lib_def(OpRegistry::Global(), graph_def.library()); in FindStatefulOps() local
Drewrite_utils.cc189 std::unique_ptr<FunctionLibraryDefinition> lib_def = nullptr; in RewriteDataset() local
Dcaptured_function.cc229 const FunctionLibraryDefinition* lib_def, const string& func_name, in CreateFunctionLibraryDefinition()
242 Status LookupFunction(const FunctionLibraryDefinition& lib_def, in LookupFunction()
/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
952 std::unique_ptr<FunctionLibraryDefinition> lib_def( in TEST() local
1144 std::unique_ptr<FunctionLibraryDefinition> lib_def( in TEST() local
1316 std::unique_ptr<FunctionLibraryDefinition> lib_def( in TEST() local
1483 std::unique_ptr<FunctionLibraryDefinition> lib_def( in TEST() local
1599 std::unique_ptr<FunctionLibraryDefinition> lib_def( in TEST() local
1696 std::unique_ptr<FunctionLibraryDefinition> lib_def( in TEST() local
1806 std::unique_ptr<FunctionLibraryDefinition> lib_def( in TEST() local
[all …]
/external/tensorflow/tensorflow/core/common_runtime/eager/
Dshape_inference.cc29 const FunctionLibraryDefinition& lib_def, in RunShapeInference()
Dkernel_and_device.cc172 const FunctionLibraryDefinition* lib_def = in InstantiateFunc() local
/external/tensorflow/tensorflow/core/distributed_runtime/eager/
Dremote_execute_node.h45 const NodeDef& ndef, FunctionLibraryDefinition* lib_def, in RemoteExecuteNode()
Dcluster_function_library_runtime.cc48 const string& function_name, const FunctionLibraryDefinition& lib_def, in Instantiate()
/external/tensorflow/tensorflow/core/grappler/optimizers/data/
Dslack_test.cc115 FunctionDefLibrary lib_def; in TEST() local
Dfunction_utils_test.cc804 FunctionLibraryDefinition lib_def(OpRegistry::Global(), in TEST() local
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ops/
Dvariable_ops.cc54 std::unique_ptr<FunctionLibraryDefinition> lib_def; in ReadVariableHelper() local
/external/tensorflow/tensorflow/core/grappler/utils/
Dfunctions_test.cc676 FunctionDefLibrary lib_def; in TEST_F() local

12