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.cc2475 FunctionLibraryDefinition function_library(OpRegistry::Global(), in InferStatically() local
/external/tensorflow/tensorflow/core/common_runtime/
Dlocal_executor_params.h41 FunctionLibraryRuntime* function_library = nullptr; member
Dgraph_runner.cc103 Status GraphRunner::Run(Graph* graph, FunctionLibraryRuntime* function_library, in Run()
Dconstant_folding.cc591 FunctionLibraryRuntime* function_library, Env* env, in ConstantFold()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_op_kernel.h258 FunctionLibraryRuntime* function_library() const { in function_library() function
/external/tensorflow/tensorflow/core/grappler/optimizers/data/
Dmap_parallelization.cc76 FunctionLibraryDefinition function_library(OpRegistry::Global(), in OptimizeAndCollectStats() local
Dbatch_parallelization.cc79 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
Dnoop_elimination.cc110 FunctionLibraryDefinition function_library(OpRegistry::Global(), in IsMapIdentity() local
Dmap_and_filter_fusion.cc177 FunctionLibraryDefinition function_library(OpRegistry::Global(), in OptimizeAndCollectStats() local
Dmap_vectorization.cc468 const FunctionLibraryDefinition& function_library, in FindMapAndBatchPattern()
543 FunctionLibraryDefinition function_library(OpRegistry::Global(), *library); in OptimizeAndCollectStats() local
Dhoist_random_uniform.cc242 FunctionLibraryDefinition function_library(OpRegistry::Global(), in OptimizeAndCollectStats() local
/external/tensorflow/tensorflow/compiler/jit/
Dxla_platform_info.cc132 const FunctionLibraryRuntime& function_library, DeviceBase* device, in GenerateCompilerOptions()
/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()
Dfunction_optimizer.cc271 const FunctionLibraryDefinition& function_library() const { in function_library() function in tensorflow::grappler::__anon36252d010111::FunctionOptimizerContext
274 FunctionLibraryDefinition& function_library() { return function_library_; } in function_library() function in tensorflow::grappler::__anon36252d010111::FunctionOptimizerContext
Dmeta_optimizer_test.cc431 std::vector<FunctionDef> function_library = {my_mul, my_fwd}; in TEST_F() local
/external/tensorflow/tensorflow/core/framework/
Dop_kernel.h349 FunctionLibraryRuntime* function_library() const { return flib_; } in function_library() function
685 FunctionLibraryRuntime* function_library = nullptr; member
1131 FunctionLibraryRuntime* function_library() const { in function_library() function
/external/tensorflow/tensorflow/core/grappler/
Dgrappler_item_builder.cc249 FunctionLibraryDefinition function_library(OpRegistry::Global(), in RuntimeGraphOptimizer() local