Home
last modified time | relevance | path

Searched defs:flib_def (Results 1 – 25 of 103) sorted by relevance

12345

/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/
H A Dresource_util_test.cc50 const GraphDef& graphdef, FunctionLibraryDefinition* flib_def, in AnalyzeAndVerify()
85 FunctionLibraryDefinition flib_def(OpRegistry::Global(), in TEST() local
120 FunctionLibraryDefinition flib_def(OpRegistry::Global(), in TEST() local
160 FunctionLibraryDefinition flib_def(OpRegistry::Global(), in TEST() local
203 FunctionLibraryDefinition flib_def(OpRegistry::Global(), in TEST() local
252 FunctionLibraryDefinition flib_def(OpRegistry::Global(), in TEST() local
315 FunctionLibraryDefinition flib_def(OpRegistry::Global(), in TEST() local
392 FunctionLibraryDefinition flib_def(OpRegistry::Global(), *library); in TEST() local
449 FunctionLibraryDefinition flib_def(OpRegistry::Global(), *library); in TEST() local
499 FunctionLibraryDefinition flib_def(OpRegistry::Global(), *library); in TEST() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/
H A Dresource_operation_safety_analysis_test.cc242 FunctionDefLibrary flib_def; in CreateFunctionDefLibWithConstFunction() local
263 FunctionDefLibrary flib_def = in TEST() local
285 FunctionDefLibrary flib_def = in TEST() local
305 FunctionDefLibrary flib_def = in TEST() local
325 FunctionDefLibrary flib_def = in TEST() local
347 FunctionDefLibrary flib_def = in TEST() local
374 FunctionDefLibrary flib_def = in TEST() local
H A Dencapsulate_xla_computations_pass_test.cc38 const FunctionLibraryDefinition& flib_def, const string& function) { in MakeOuterGraph()
148 FunctionLibraryDefinition flib_def(OpRegistry::Global(), {}); in TEST() local
199 FunctionLibraryDefinition flib_def(OpRegistry::Global(), {}); in TEST() local
298 FunctionLibraryDefinition flib_def(OpRegistry::Global(), flib); in TEST() local
H A Dmark_for_compilation_pass_test_helper.cc25 std::unique_ptr<Graph>* graph, FunctionLibraryDefinition* flib_def, in MarkForCompilation()
70 FunctionLibraryDefinition flib_def((*graph)->op_registry(), flib); in MarkForCompilation() local
H A Dforce_xla_constants_on_host_pass_test.cc41 FunctionLibraryDefinition* flib_def, in ForceXlaConstantsOnHost()
80 FunctionLibraryDefinition flib_def(OpRegistry::Global(), library); in TEST() local
H A Dbuild_xla_ops_pass.cc298 const FunctionLibraryDefinition& flib_def, Node* n, Graph* g, in ReplaceFunctionCallWithPartitionedCall()
348 const string& function_name, const FunctionLibraryDefinition& flib_def) { in InferDeviceForCluster()
469 const FunctionLibraryDefinition& flib_def, bool lazy_compilation_enabled, in ReplaceNodeWithXlaCompileAndXlaRun()
H A Dresource_operation_safety_analysis.cc101 const Node& n, const FunctionLibraryDefinition* flib_def, in XlaResourceOpKindForNode()
250 const Graph& g, const FunctionLibraryDefinition* flib_def, in ComputeIncompatibleResourceOperationPairs()
H A Dxla_cluster_util_test.cc139 const Scope& scope, FunctionLibraryDefinition* flib_def = nullptr) { in GetNodesRelatedToRefVarsSorted()
240 FunctionLibraryDefinition flib_def(OpRegistry::Global(), fdef_lib); in TEST() local
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/
H A Dgraph_execution_state.cc75 std::unique_ptr<FunctionLibraryDefinition>&& flib_def, in GraphExecutionState()
97 auto flib_def = std::make_unique<FunctionLibraryDefinition>( in MakeForBaseGraph() local
161 auto flib_def = std::make_unique<FunctionLibraryDefinition>( in MakeForPrunedGraph() local
267 auto flib_def = std::make_unique<FunctionLibraryDefinition>( in Extend() local
647 const FunctionLibraryDefinition* flib_def, in OptimizeGraph()
H A Dpartitioning_utils_test.cc72 FunctionLibraryDefinition flib_def(OpRegistry::Global()); in SwapGraph() local
89 FunctionLibraryDefinition flib_def(OpRegistry::Global()); in TwoDeviceSwapGraph() local
110 FunctionLibraryDefinition flib_def(OpRegistry::Global()); in SubGraph() local
175 FunctionLibraryDefinition flib_def(OpRegistry::Global()); in TEST_F() local
220 FunctionLibraryDefinition flib_def(OpRegistry::Global()); in TEST_F() local
H A Dplacer.cc145 const FunctionLibraryDefinition* flib_def, in Placer()
157 const FunctionLibraryDefinition* flib_def, in Placer()
162 const FunctionLibraryDefinition* flib_def, in Placer()
H A Disolate_placer_inspection_required_ops_pass_test.cc42 FunctionLibraryDefinition* flib_def) { in RunPass()
55 FunctionLibraryDefinition flib_def(OpRegistry::Global(), original.library()); in RunPass() local
156 FunctionLibraryDefinition flib_def(OpRegistry::Global(), {}); in TEST() local
H A Dplacer_inspection_required_ops_utils.cc51 const Graph* graph, const FunctionLibraryDefinition* flib_def) in PlacerInspectionRequiredOpChecker()
80 Status GetFunctionDefAndAttrs(const FunctionLibraryDefinition& flib_def, in GetFunctionDefAndAttrs()
287 const FunctionLibraryDefinition& flib_def, Graph* graph) { in IsolatePlacerInspectionRequiredOps()
H A Dlower_while_op.cc67 Graph* graph, const FunctionLibraryDefinition* flib_def, in Run()
183 const FunctionLibraryDefinition* flib_def, in LowerWhileHelper()
508 const FunctionLibraryDefinition* flib_def, in RewriteWhileNode()
H A Dgraph_execution_state.h65 std::unique_ptr<FunctionLibraryDefinition> flib_def; member
155 const FunctionLibraryDefinition& flib_def() const { return *flib_def_; } in flib_def() function
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/
H A Diterator_ops.cc93 std::unique_ptr<FunctionLibraryDefinition> flib_def, in IteratorResource()
468 std::unique_ptr<FunctionLibraryDefinition> flib_def(nullptr); in Compute() local
523 std::unique_ptr<FunctionLibraryDefinition>* flib_def, in CreatePrivateFLR()
571 OpKernelContext* ctx, std::unique_ptr<FunctionLibraryDefinition> flib_def, in CreateResource()
788 std::unique_ptr<FunctionLibraryDefinition> flib_def(nullptr); in TryInit() local
H A Diterator_ops.h83 State(std::shared_ptr<FunctionLibraryDefinition> flib_def, in State()
106 std::shared_ptr<FunctionLibraryDefinition> flib_def() { return flib_def_; } in flib_def() function
H A Dmulti_device_iterator_ops.cc81 std::unique_ptr<FunctionLibraryDefinition> flib_def, in MultiDeviceIterator()
514 std::unique_ptr<FunctionLibraryDefinition> flib_def(nullptr); in Compute() local
614 std::unique_ptr<FunctionLibraryDefinition> flib_def, in CreateResource()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/
H A Dexport_graphdef.cc668 FunctionLibraryDefinition* flib_def, in Convert()
723 FunctionLibraryDefinition* flib_def, in ConvertMlirToGraph()
734 FunctionLibraryDefinition* flib_def) { in ConvertMlirToGraph()
742 FunctionLibraryDefinition flib_def(OpRegistry::Global(), in ConvertMlirToGraphdef() local
H A Dupgrade_graph.cc79 auto flib_def = library ? std::make_unique<FunctionLibraryDefinition>( in GenerateResourceSharedNameIfEmpty() local
131 Status UpgradeLegacyGraph(Graph* graph, FunctionLibraryDefinition* flib_def, in UpgradeLegacyGraph()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
H A Ddump_graph.cc65 const FunctionLibraryDefinition* flib_def, in DumpTextualIRToFile()
101 WritableFile* file) -> Status { in UseMlirForGraphDump()
H A Dcompile_mlir_util.cc843 const FunctionLibraryDefinition& flib_def, const GraphDebugInfo& debug_info, in GraphToModule()
865 const FunctionLibraryDefinition& flib_def, const GraphDebugInfo& debug_info, in BuildHloFromGraph()
879 const FunctionLibraryDefinition& flib_def, const GraphDebugInfo& debug_info, in CompileGraphToXlaHlo()
/aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/utils/
H A Dgraph_partition_test.cc85 FunctionLibraryDefinition flib_def(OpRegistry::Global()); in TEST_F() local
134 FunctionLibraryDefinition flib_def(OpRegistry::Global()); in TEST_F() local
/aosp_15_r20/external/tensorflow/tensorflow/core/data/
H A Dstandalone.cc77 auto flib_def = std::make_unique<FunctionLibraryDefinition>( in FromGraph() local
169 FunctionLibraryDefinition* flib_def, thread::ThreadPool* pool, in Dataset()
/aosp_15_r20/external/tensorflow/tensorflow/core/util/
H A Ddump_graph.cc214 const FunctionLibraryDefinition* flib_def, in SetGraphDumper()
239 const FunctionLibraryDefinition* flib_def, in DumpGraphToFile()

12345