Home
last modified time | relevance | path

Searched defs:function_library (Results 1 – 22 of 22) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/verifiers/
Dstructure_verifier.cc37 FunctionLibraryDefinition function_library(OpRegistry::Global(), in Verify() local
/external/tensorflow/tensorflow/core/grappler/costs/
Dop_context.h31 const FunctionDefLibrary* function_library; // Not owned. member
Dgraph_properties.cc2040 FunctionLibraryDefinition function_library(OpRegistry::Global(), in InferStatically() local
/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_op_kernel.h230 FunctionLibraryRuntime* function_library() const { in function_library() function
/external/tensorflow/tensorflow/core/grappler/optimizers/data/
Dmap_parallelization.cc62 FunctionLibraryDefinition function_library(OpRegistry::Global(), in OptimizeAndCollectStats() local
Dfilter_fusion.cc71 FunctionLibraryDefinition function_library(OpRegistry::Global(), in OptimizeAndCollectStats() local
Dmap_fusion.cc91 FunctionLibraryDefinition function_library(OpRegistry::Global(), in OptimizeAndCollectStats() local
Dmap_and_filter_fusion.cc108 FunctionLibraryDefinition function_library(OpRegistry::Global(), in OptimizeAndCollectStats() local
Dmap_vectorization.cc447 const FunctionLibraryDefinition& function_library, in FindMapAndBatchPattern()
521 FunctionLibraryDefinition function_library(OpRegistry::Global(), *library); in OptimizeAndCollectStats() local
Dhoist_random_uniform.cc232 FunctionLibraryDefinition function_library(OpRegistry::Global(), in OptimizeAndCollectStats() local
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dfunction_optimizer_test.cc1437 std::vector<FunctionDef> function_library = {x_times_two}; in TEST_F() local
1486 std::vector<FunctionDef> function_library = {x_times_two}; in TEST_F() local
1554 std::vector<FunctionDef> function_library = {mul_func}; in TEST_F() local
1618 std::vector<FunctionDef> function_library = {mul_func}; in TEST_F() local
1696 std::vector<FunctionDef> function_library = {mul_func}; in TEST_F() local
1820 std::vector<FunctionDef> function_library = {my_func}; in TEST_F() local
1928 std::vector<FunctionDef> function_library = {my_func}; in TEST_F() local
Dfunction_api_info.cc157 const FunctionDefLibrary& function_library) { in Init()
Dmeta_optimizer_test.cc433 std::vector<FunctionDef> function_library = {my_mul, my_fwd}; in TEST_F() local
Dfunction_optimizer.cc288 const FunctionLibraryDefinition& function_library() const { in function_library() function in tensorflow::grappler::__anon35816ea30111::FunctionOptimizerContext
/external/tensorflow/tensorflow/compiler/jit/
Dxla_fusion_optimizer.cc136 FunctionLibraryDefinition function_library(OpRegistry::Global(), in Optimize() local
/external/tensorflow/tensorflow/core/common_runtime/
Dgraph_runner.cc101 Status GraphRunner::Run(Graph* graph, FunctionLibraryRuntime* function_library, in Run()
Dexecutor.h131 FunctionLibraryRuntime* function_library = nullptr; member
Dconstant_folding.cc564 FunctionLibraryRuntime* function_library, Env* env, in ConstantFold()
/external/tensorflow/tensorflow/core/framework/
Ddataset.h321 std::shared_ptr<const FunctionLibraryDefinition> function_library = nullptr; member
366 std::shared_ptr<const FunctionLibraryDefinition> function_library() { in function_library() function
Dop_kernel.h353 FunctionLibraryRuntime* function_library() const { return flib_; } in function_library() function
667 FunctionLibraryRuntime* function_library = nullptr; member
1089 FunctionLibraryRuntime* function_library() const { in function_library() function
/external/tensorflow/tensorflow/core/grappler/
Dgrappler_item_builder.cc115 FunctionLibraryDefinition function_library(OpRegistry::Global(), in OptimizeGraph() local
/external/tensorflow/tensorflow/core/kernels/data/
Dmulti_device_iterator_ops.cc114 std::shared_ptr<const FunctionLibraryDefinition> function_library() { in function_library() function in tensorflow::data::__anon2db6daca0111::MultiDeviceIterator