| /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()
69 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 | 37 const FunctionLibraryDefinition& flib_def, const string& function) { in MakeOuterGraph()
147 FunctionLibraryDefinition flib_def(OpRegistry::Global(), {}); in TEST() local
205 FunctionLibraryDefinition flib_def(OpRegistry::Global(), {}); in TEST() local
304 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/utils/ |
| D | dump_graph.cc | 65 const FunctionLibraryDefinition* flib_def, in DumpTextualIRToFile()
99 WritableFile* file) -> Status { in UseMlirForGraphDump()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | graph_execution_state.h | 65 std::unique_ptr<FunctionLibraryDefinition> flib_def; member
150 const FunctionLibraryDefinition& flib_def() const { return *flib_def_; } in flib_def() function
|
| D | partitioning_utils.h | 74 FunctionNameGenerator(const FunctionLibraryDefinition* flib_def, in FunctionNameGenerator()
|
| D | isolate_placer_inspection_required_ops_pass.cc | 41 const FunctionLibraryDefinition* flib_def = in Run() local
|
| D | graph_execution_state.cc | 74 std::unique_ptr<FunctionLibraryDefinition>&& flib_def, in GraphExecutionState()
96 auto flib_def = absl::make_unique<FunctionLibraryDefinition>( in MakeForBaseGraph() local
160 auto flib_def = absl::make_unique<FunctionLibraryDefinition>( in MakeForPrunedGraph() local
266 auto flib_def = absl::make_unique<FunctionLibraryDefinition>( in Extend() local
|
| D | placer_inspection_required_ops_utils.cc | 55 const Graph* graph, const FunctionLibraryDefinition* flib_def) in PlacerInspectionRequiredOpChecker()
84 Status GetFunctionDefAndAttrs(const FunctionLibraryDefinition& flib_def, in GetFunctionDefAndAttrs()
299 const FunctionLibraryDefinition& flib_def, Graph* graph) { in IsolatePlacerInspectionRequiredOps()
|
| D | function_optimization_registry.cc | 28 std::unique_ptr<Graph>* graph, FunctionLibraryDefinition* flib_def, in Run()
|
| D | function_optimization_registry_pass_failure_test.cc | 32 std::unique_ptr<Graph>* graph, FunctionLibraryDefinition* flib_def, in Run()
|
| D | function_optimization_registration_test.cc | 31 std::unique_ptr<Graph>* graph, FunctionLibraryDefinition* flib_def, in Run()
|
| D | function_optimization_registry_test.cc | 33 std::unique_ptr<Graph>* graph, FunctionLibraryDefinition* flib_def, in Run()
|
| D | lower_function_call_op.cc | 46 const FunctionLibraryDefinition& flib_def, in RewriteFunctionCallNode()
|
| D | isolate_placer_inspection_required_ops_pass_test.cc | 42 FunctionLibraryDefinition* flib_def = nullptr) { in RunPass()
152 FunctionLibraryDefinition flib_def(OpRegistry::Global(), {}); in TEST() 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/compiler/mlir/tensorflow/translate/ |
| D | upgrade_graph.cc | 74 auto flib_def = library ? std::make_unique<FunctionLibraryDefinition>( in GenerateResourceSharedNameIfEmpty() local
180 Status UpgradeLegacyGraph(Graph* graph, FunctionLibraryDefinition* flib_def, in UpgradeLegacyGraph()
|
| D | export_graphdef.cc | 620 FunctionLibraryDefinition flib_def(&empty_registry, *flib); in ConvertLibFunction() local
706 FunctionLibraryDefinition* flib_def, in Convert()
745 FunctionLibraryDefinition* flib_def, in ConvertMlirToGraph()
754 FunctionLibraryDefinition* flib_def) { in ConvertMlirToGraph()
762 FunctionLibraryDefinition flib_def(OpRegistry::Global(), in ConvertMlirToGraphdef() local
|
| /external/tensorflow/tensorflow/core/data/ |
| D | standalone.cc | 56 auto flib_def = in FromGraph() local
146 FunctionLibraryDefinition* flib_def, thread::ThreadPool* pool) in Dataset()
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | iterator_ops.h | 78 State(std::shared_ptr<FunctionLibraryDefinition> flib_def, in State()
96 std::shared_ptr<FunctionLibraryDefinition> flib_def() { return flib_def_; } in flib_def() function
|
| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| D | tpu_compile_op_support.h | 60 const FunctionLibraryDefinition* const flib_def; member
|
| /external/tensorflow/tensorflow/compiler/mlir/tfjs/translate/ |
| D | json_translate.cc | 61 tensorflow::FunctionLibraryDefinition flib_def( in MlirToJSONTranslateFunction() local
|