| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | optimization_registry_test.cc | 61 flib_def_.reset( in OptimizationPassTest()
67 options.flib_def = flib_def_.get(); in RunPass()
74 return flib_def_->Find(func); in GetFunctionDef()
78 std::unique_ptr<FunctionLibraryDefinition> flib_def_; member in tensorflow::OptimizationPassTest
|
| D | inspecting_placer.cc | 120 flib_def_(*flib_def), in InspectingPlacer()
130 TF_RETURN_IF_ERROR(GetFunctionDefAndAttrs(flib_def_, node, &fdef, &func)); in ComputeIOColocationGroups()
134 &flib_def_, &fbody)); in ComputeIOColocationGroups()
137 IsolatePlacerInspectionRequiredOps(flib_def_, fbody->graph)); in ComputeIOColocationGroups()
149 fbody->graph, stack_.Push(&node, func.name()), &flib_def_, &device_set_, in ComputeIOColocationGroups()
|
| D | partitioning_utils.h | 76 : flib_def_(flib_def), name_(name), counter_(0) {} in FunctionNameGenerator()
83 const FunctionLibraryDefinition* flib_def_;
|
| D | graph_execution_state.cc | 81 flib_def_(std::move(flib_def)), in GraphExecutionState()
192 TF_RETURN_IF_ERROR(flib_def_->AddLibrary(extension_def.library())); in Extend()
193 *gdef.mutable_library() = flib_def_->ToProto(); in Extend()
219 AddDefaultAttrsToGraphDef(&gdef, *flib_def_, old_node_size)); in Extend()
255 TF_RETURN_IF_ERROR(graph::ValidateGraphDef(gdef, *flib_def_)); in Extend()
265 TF_RETURN_IF_ERROR(AddDefaultAttrsToGraphDef(&gdef, *flib_def_, 0)); in Extend()
614 optimization_options.flib_def = flib_def_.get(); in InitBaseGraph()
620 Placer placer(new_graph.get(), "", flib_def_.get(), device_set_, in InitBaseGraph()
787 if (flib_def_) { in OptimizeGraph()
788 *item.graph.mutable_library() = flib_def_->ToProto(); in OptimizeGraph()
[all …]
|
| D | graph_execution_state.h | 150 const FunctionLibraryDefinition& flib_def() const { return *flib_def_; } in flib_def()
208 std::unique_ptr<FunctionLibraryDefinition> flib_def_; variable
|
| D | inspecting_placer.h | 84 const FunctionLibraryDefinition& flib_def_; variable
|
| D | placer.h | 98 const FunctionLibraryDefinition* const flib_def_; // Not owned. variable
|
| D | placer_inspection_required_ops_utils.h | 68 const FunctionLibraryDefinition& flib_def_; variable
|
| D | placer.cc | 140 flib_def_(flib_def), in Placer()
175 ColocationGraph colocation_graph(graph_, stack, flib_def_, devices_, in Run()
|
| D | kernel_benchmark_testlib.h | 94 std::unique_ptr<FunctionLibraryDefinition> flib_def_; variable
|
| D | kernel_benchmark_testlib.cc | 90 flib_def_ = absl::make_unique<FunctionLibraryDefinition>(g->flib_def()); in Benchmark()
95 flib_def_.get(), OptimizerOptions(), pool_, nullptr, nullptr, in Benchmark()
|
| D | placer_inspection_required_ops_utils.cc | 56 : graph_(*graph), flib_def_(*flib_def) { in PlacerInspectionRequiredOpChecker()
72 TF_RETURN_IF_ERROR(GetFunctionDefAndAttrs(flib_def_, node, &fdef, &func)); in IsPlacerInspectionRequired()
|
| D | colocation_graph.h | 370 const FunctionLibraryDefinition& flib_def_; variable
|
| D | partitioning_utils.cc | 164 if (flib_def_->Find(candidate) == nullptr) { in GetName()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | compilability_check_util_test.cc | 100 flib_def_.get(), opts); in GetFunctionLibraryRuntime()
109 std::unique_ptr<FunctionLibraryDefinition> flib_def_ = member in tensorflow::__anon903eb31b0111::CompilabilityCheckUtilTest
158 flib_def_.reset(new FunctionLibraryDefinition(OpRegistry::Global(), flib)); in TEST_F()
160 GraphDefBuilder builder(GraphDefBuilder::kFailImmediately, flib_def_.get()); in TEST_F()
161 std::unique_ptr<Graph> graph(new Graph(flib_def_.get())); in TEST_F()
210 flib_def_.reset(new FunctionLibraryDefinition(OpRegistry::Global(), flib)); in TEST_F()
211 GraphDefBuilder builder(GraphDefBuilder::kFailImmediately, flib_def_.get()); in TEST_F()
230 std::unique_ptr<Graph> graph(new Graph(flib_def_.get())); in TEST_F()
309 flib_def_.reset(new FunctionLibraryDefinition(OpRegistry::Global(), flib)); in TEST_F()
406 flib_def_.reset(new FunctionLibraryDefinition(OpRegistry::Global(), flib)); in TEST_F()
|
| D | mark_for_compilation_pass.cc | 116 flib_def_(flib_def), in MarkForCompilationPassImpl()
426 FunctionLibraryDefinition* flib_def_; member in tensorflow::__anonc4d28a150111::MarkForCompilationPassImpl
855 DumpGraphToFile("before_mark_for_compilation", *graph_, flib_def_); in CreateClusters()
998 *graph_, flib_def_, ignore_resource_ops, &unsafe_resource_deps_vect)); in BuildInitialClusterSet()
1040 if (is_resource_op || MayCallFunction(*node, flib_def_)) { in BuildInitialClusterSet()
1045 GetNodeOrFuncAttr(node, flib_def_, kXlaCompileAttr) || in BuildInitialClusterSet()
1047 GetNodeOrFuncAttr(node, flib_def_, kXlaMustCompileAttr)); in BuildInitialClusterSet()
1132 TF_GRAPH_DEF_VERSION, flib_def_, opts)); in FindCompilationCandidates()
1333 status = flib_def_->GetAttr(*node, kXlaCompileAttr, &compile); in CompilationDisallowedByXlaCompileAttr()
1445 DumpGraphToFile("mark_for_compilation", *graph_, flib_def_); in DumpPostClusteringGraphs()
[all …]
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | kernel_and_device_test.cc | 44 TestEnv() : flib_def_(OpRegistry::Global(), {}) { in TestEnv()
53 TF_GRAPH_DEF_VERSION, &flib_def_, opts,
65 FunctionLibraryDefinition flib_def_; member in tensorflow::__anond3bba5f70111::TestEnv
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | functionalize_cond_test.cc | 41 flib_def_.reset( in FunctionalizeCondTest()
44 graph_.get(), flib_def_.get(), NodeFilter{})); in FunctionalizeCondTest()
68 std::unique_ptr<FunctionLibraryDefinition> flib_def_; member in tensorflow::functionalize_cond::FunctionalizeCondTest
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | ops_testutil.cc | 83 flib_def_ = absl::make_unique<FunctionLibraryDefinition>( in OpsTestBase()
87 TF_GRAPH_DEF_VERSION, flib_def_.get(), OptimizerOptions()); in OpsTestBase()
111 TF_GRAPH_DEF_VERSION, flib_def_.get(), OptimizerOptions(), in SetDevice()
|
| D | ops_testutil.h | 185 std::unique_ptr<FunctionLibraryDefinition> flib_def_; variable
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | iterator_ops.h | 82 : flib_def_(std::move(flib_def)), in State()
96 std::shared_ptr<FunctionLibraryDefinition> flib_def() { return flib_def_; } in flib_def()
115 std::shared_ptr<FunctionLibraryDefinition> flib_def_;
|
| /external/tensorflow/tensorflow/core/data/ |
| D | standalone.h | 117 std::unique_ptr<FunctionLibraryDefinition> flib_def_; variable
|
| D | standalone.cc | 149 flib_def_(flib_def), in Dataset()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf_with_tf2xla.cc | 349 std::unique_ptr<tensorflow::FunctionLibraryDefinition> flib_def_; member in mlir::mhlo::__anon94f199800111::Tf2XlaRewriter
394 flib_def_ = absl::make_unique<tensorflow::FunctionLibraryDefinition>( in PrepareParams()
398 version_or.ValueOrDie(), flib_def_.get(), tensorflow::OptimizerOptions()); in PrepareParams()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/kernels/ |
| D | trt_engine_op_test.cc | 83 TF_ASSERT_OK(flib_def_->AddLibrary(graph->flib_def()));
|