Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dgraph_optimization_pass.h34 const tensorflow::FunctionLibraryDefinition& function_library) in GetPassState()
Dgraph_optimization_pass.cc36 const tensorflow::FunctionLibraryDefinition& function_library) { in Run()
/external/tensorflow/tensorflow/core/grappler/verifiers/
Dstructure_verifier.cc37 FunctionLibraryDefinition function_library(OpRegistry::Global(), in Verify() local
/external/tensorflow/tensorflow/compiler/mlir/
Dmlir_bridge_rollout_policy.cc24 const FunctionLibraryDefinition* function_library, in GetMlirBridgeRolloutPolicy()
Dmlir_graph_optimization_pass_test.cc75 const FunctionLibraryDefinition& function_library) override { in Run()
/external/tensorflow/tensorflow/core/grappler/costs/
Dop_context.h31 const FunctionDefLibrary* function_library; // Not owned. member
/external/tensorflow/tensorflow/cc/experimental/libexport/
Dload.cc67 auto& function_library = in GetFunctionDefs() local
/external/tensorflow/tensorflow/compiler/mlir/tfr/integration/
Dgraph_decompose_pass.cc44 const FunctionLibraryDefinition& function_library) { in Run()
/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.h303 FunctionLibraryRuntime* function_library() const { in function_library() function
Dmlir_bridge_pass.cc122 const FunctionLibraryDefinition& function_library) { in Run()
/external/tensorflow/tensorflow/core/grappler/optimizers/data/
Dmap_parallelization.cc76 FunctionLibraryDefinition function_library(OpRegistry::Global(), in OptimizeAndCollectStats() local
Dbatch_parallelization.cc77 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.cc120 FunctionLibraryDefinition function_library(OpRegistry::Global(), in IsMapIdentity() local
Dmap_and_filter_fusion.cc177 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()
Dfunction_optimizer.cc274 const FunctionLibraryDefinition& function_library() const { in function_library() function in tensorflow::grappler::__anon42ddc3660111::FunctionOptimizerContext
277 FunctionLibraryDefinition& function_library() { return function_library_; } in function_library() function in tensorflow::grappler::__anon42ddc3660111::FunctionOptimizerContext
/external/tensorflow/tensorflow/compiler/jit/
Dxla_platform_info.cc158 const FunctionLibraryRuntime& function_library, DeviceBase* device, in GenerateCompilerOptions()
/external/tensorflow/tensorflow/core/framework/
Dop_kernel.h306 FunctionLibraryRuntime* function_library() const { return flib_; } in function_library() function
645 FunctionLibraryRuntime* function_library = nullptr; member
1076 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

12