| /aosp_15_r20/external/tensorflow/tensorflow/core/grappler/utils/ |
| H A D | functions_test.cc | 128 FunctionLibraryDefinition flib(OpRegistry::Global(), FunctionDefLibrary()); in TEST_F() local
209 FunctionLibraryDefinition flib(OpRegistry::Global(), FunctionDefLibrary()); in TEST_F() local
268 FunctionLibraryDefinition flib(OpRegistry::Global(), FunctionDefLibrary()); in TEST_F() local
369 FunctionLibraryDefinition flib(OpRegistry::Global(), FunctionDefLibrary()); in TEST_F() local
421 FunctionLibraryDefinition flib(OpRegistry::Global(), FunctionDefLibrary()); in TEST_F() local
460 FunctionLibraryDefinition flib(OpRegistry::Global(), FunctionDefLibrary()); in TEST_F() local
487 FunctionLibraryDefinition flib(OpRegistry::Global(), FunctionDefLibrary()); in TEST_F() local
536 FunctionLibraryDefinition flib(OpRegistry::Global(), FunctionDefLibrary()); in TEST_F() local
585 FunctionLibraryDefinition flib(OpRegistry::Global(), FunctionDefLibrary()); in TEST_F() local
679 FunctionLibraryDefinition flib(OpRegistry::Global(), lib_def); in TEST_F() local
[all …]
|
| H A D | functions.cc | 223 const FunctionLibraryDefinition& flib, in MakeGrapplerFunctionItem()
307 const FunctionLibraryDefinition& flib, in MakeGrapplerFunctionItem()
448 const GrapplerFunctionItem& item, const FunctionLibraryDefinition& flib) { in Initialize()
524 const FunctionLibraryDefinition& flib, in MakeFunctionDef()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/ |
| H A D | export_graphdef.cc | 425 const SymbolTable& symbol_table, FuncOp function, FunctionDefLibrary* flib, in Convert()
590 const SymbolTable& symbol_table, FuncOp function, FunctionDefLibrary* flib, in ConvertLibFunction()
673 FunctionDefLibrary flib; in Convert() local
752 FunctionDefLibrary flib = flib_def.ToProto(); in ConvertMlirToGraphdef() local
775 FunctionDefLibrary flib; in ConvertMlirFunctionToFunctionLibraryDef() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/data/ |
| H A D | auto_shard.cc | 348 const FunctionLibraryDefinition& flib) { in ReaderOpInFunction()
483 FunctionLibraryDefinition* flib, MutableGraphView* graph, in FindFuncAndTensorSliceDataset()
543 int64_t index, FunctionLibraryDefinition* flib, in RecursivelyHandleOp()
648 FunctionLibraryDefinition* flib, MutableGraphView* graph) { in ShardByFile()
761 FunctionLibraryDefinition flib(OpRegistry::Global(), in ApplyAutoShard() local
|
| H A D | meta_optimizer.cc | 118 FunctionLibraryDefinition flib = in Optimize() local
|
| H A D | slack_test.cc | 156 FunctionLibraryDefinition flib(OpRegistry::Global(), lib_def); in TEST() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/ |
| H A D | compilability_check_util_test.cc | 180 FunctionDefLibrary flib; in TEST_F() local
221 FunctionDefLibrary flib; in TEST_F() local
301 FunctionDefLibrary flib; in TEST_F() local
402 FunctionDefLibrary flib; in TEST_F() local
|
| H A D | mark_for_compilation_pass_test_helper.cc | 69 FunctionDefLibrary flib; in MarkForCompilation() local
|
| H A D | xla_kernel_creator_test.cc | 61 void Init(const std::vector<FunctionDef>& flib) { in Init()
|
| H A D | mark_for_compilation_pass_test.cc | 247 FunctionDefLibrary flib; in TEST() local
287 FunctionDefLibrary flib; in TEST() local
326 FunctionDefLibrary flib; in TEST() local
503 FunctionDefLibrary flib; in TEST() local
|
| H A D | xla_cluster_util_test.cc | 140 FunctionDefLibrary flib; in GetNodesRelatedToRefVarsSorted() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | batch_kernels.cc | 139 FunctionLibraryRuntime* flib, in Create()
162 FunctionLibraryRuntime::Handle fhandle, FunctionLibraryRuntime* flib, in Create()
181 FunctionLibraryRuntime* flib, std::shared_ptr<BatcherT> batcher, in BatchResource()
192 FunctionLibraryRuntime* flib, in BatchResource()
|
| H A D | partitioned_function_ops.cc | 138 const FunctionLibraryDefinition* flib = lib.GetFunctionLibraryDefinition(); in FillOutputDevices() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/ |
| H A D | meta_optimizer.cc | 983 void PropagateTFDataAttrs(const FunctionLibraryDefinition& flib, in PropagateTFDataAttrs()
1082 FunctionLibraryDefinition flib = minimized_flib(*optimized_graph); in OptimizeConsumeItem() local
1369 FunctionLibraryDefinition* flib, const DeviceSet& device_set, in OptimizeGraph()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/fallback/ |
| H A D | fallback_state_test.cc | 34 FunctionDefLibrary flib; in TEST() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/ |
| H A D | graph_compiler.h | 59 FunctionLibraryRuntime* flib, in GraphCompiler()
|
| H A D | xla_compiler_test.cc | 68 FunctionDefLibrary flib; in SetUp() local
967 FunctionDefLibrary flib; in TEST_F() local
1051 FunctionDefLibrary flib; in TEST_F() local
1480 FunctionDefLibrary flib; in TEST_F() local
|
| H A D | graph_compiler.cc | 201 Status GetFunctionNameAndAttr(const FunctionLibraryRuntime& flib, in GetFunctionNameAndAttr()
|
| H A D | const_analysis_test.cc | 101 FunctionDefLibrary flib; in TestFunctionCall() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| H A D | trt_layout_optimization_pass.cc | 66 FunctionLibraryDefinition flib(OpRegistry::Global(), graph_def.library()); in Optimize() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/tools/optimization/ |
| H A D | optimization_pass_runner.cc | 100 FunctionDefLibrary flib; in Run() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/data/ |
| H A D | hash_utils.cc | 149 const FunctionLibraryDefinition* flib) in GraphHasher()
156 const FunctionLibraryDefinition* flib, in GraphHasher()
|
| H A D | hash_utils_test.cc | 631 FunctionLibraryDefinition flib(OpRegistry::Global(), gd.library()); in TEST_F() local
679 FunctionLibraryDefinition flib(OpRegistry::Global(), gd.library()); in TEST_F() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/framework/ |
| H A D | op_kernel.cc | 259 FunctionLibraryRuntime* flib, ResourceMgr* resource_mgr, in OpKernelConstruction()
1671 Allocator* allocator, FunctionLibraryRuntime* flib, in CreateOpKernel()
1680 Allocator* allocator, FunctionLibraryRuntime* flib, in CreateOpKernel()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/ |
| H A D | function_threadpool_test.cc | 51 void Init(const std::vector<FunctionDef>& flib, in Init()
|