| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | resource_util_test.cc | 50 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
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | mark_for_compilation_pass_test_helper.cc | 25 std::unique_ptr<Graph>* graph, FunctionLibraryDefinition* flib_def, in MarkForCompilation()
70 FunctionLibraryDefinition flib_def((*graph)->op_registry(), flib); in MarkForCompilation() local
|
| D | force_xla_constants_on_host_pass_test.cc | 41 FunctionLibraryDefinition* flib_def, in ForceXlaConstantsOnHost()
80 FunctionLibraryDefinition flib_def(OpRegistry::Global(), library); in TEST() local
|
| D | resource_operation_safety_analysis_test.cc | 242 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
|
| D | encapsulate_xla_computations_pass_test.cc | 38 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
|
| D | resource_operation_safety_analysis.cc | 101 const Node& n, const FunctionLibraryDefinition* flib_def, in XlaResourceOpKindForNode()
250 const Graph& g, const FunctionLibraryDefinition* flib_def, in ComputeIncompatibleResourceOperationPairs()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/ |
| D | upgrade_graph.cc | 79 auto flib_def = library ? std::make_unique<FunctionLibraryDefinition>( in GenerateResourceSharedNameIfEmpty() local
131 Status UpgradeLegacyGraph(Graph* graph, FunctionLibraryDefinition* flib_def, in UpgradeLegacyGraph()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/ |
| D | dump_graph.cc | 65 const FunctionLibraryDefinition* flib_def, in DumpTextualIRToFile()
101 WritableFile* file) -> Status { in UseMlirForGraphDump()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | placer.cc | 145 const FunctionLibraryDefinition* flib_def, in Placer()
157 const FunctionLibraryDefinition* flib_def, in Placer()
162 const FunctionLibraryDefinition* flib_def, in Placer()
|
| D | graph_execution_state.cc | 75 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()
|
| D | graph_execution_state.h | 65 std::unique_ptr<FunctionLibraryDefinition> flib_def; member
155 const FunctionLibraryDefinition& flib_def() const { return *flib_def_; } in flib_def() function
|
| D | isolate_placer_inspection_required_ops_pass_test.cc | 42 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
|
| D | partitioning_utils_test.cc | 72 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
|
| D | partitioning_utils.h | 87 FunctionNameGenerator(const FunctionLibraryDefinition* flib_def, in FunctionNameGenerator()
|
| D | placer_inspection_required_ops_utils.cc | 51 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()
|
| D | lower_while_op.cc | 67 Graph* graph, const FunctionLibraryDefinition* flib_def, in Run()
183 const FunctionLibraryDefinition* flib_def, in LowerWhileHelper()
508 const FunctionLibraryDefinition* flib_def, in RewriteWhileNode()
|
| D | function_optimization_registry.cc | 30 std::unique_ptr<Graph>* graph, FunctionLibraryDefinition* flib_def, in Run()
|
| D | lower_function_call_op.cc | 33 const FunctionLibraryDefinition& flib_def, in RewriteFunctionCallNode()
|
| D | function_optimization_registry_test.cc | 33 std::unique_ptr<Graph>* graph, FunctionLibraryDefinition* flib_def, in Run()
|
| D | inline_function_utils_test.cc | 38 FunctionLibraryDefinition flib_def(OpRegistry::Global(), {}); in TEST() local
|
| D | function_optimization_registration_test.cc | 31 std::unique_ptr<Graph>* graph, FunctionLibraryDefinition* flib_def, in Run()
|
| /external/tensorflow/tensorflow/core/tfrt/utils/ |
| D | graph_partition_test.cc | 85 FunctionLibraryDefinition flib_def(OpRegistry::Global()); in TEST_F() local
134 FunctionLibraryDefinition flib_def(OpRegistry::Global()); in TEST_F() local
|
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| D | add_default_attributes.cc | 28 std::unique_ptr<FunctionLibraryDefinition> flib_def( in AddDefaultAttributes() local
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | iterator_ops.h | 83 State(std::shared_ptr<FunctionLibraryDefinition> flib_def, in State()
106 std::shared_ptr<FunctionLibraryDefinition> flib_def() { return flib_def_; } in flib_def() function
|
| /external/tensorflow/tensorflow/core/data/ |
| D | standalone.cc | 77 auto flib_def = std::make_unique<FunctionLibraryDefinition>( in FromGraph() local
169 FunctionLibraryDefinition* flib_def, thread::ThreadPool* pool, in Dataset()
|