| /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 | graph_execution_state.cc | 65 flib_def_(new FunctionLibraryDefinition(OpRegistry::Global(), in GraphExecutionState()
92 AddDefaultAttrsToGraphDef(&ret->original_graph_def_, *ret->flib_def_, 0)); in MakeForBaseGraph()
121 AddDefaultAttrsToGraphDef(&ret->original_graph_def_, *ret->flib_def_, 0)); in MakeForPrunedGraph()
134 TF_RETURN_IF_ERROR(flib_def_->AddLibrary(extension_def.library())); in Extend()
135 *gdef.mutable_library() = flib_def_->ToProto(); in Extend()
161 AddDefaultAttrsToGraphDef(&gdef, *flib_def_, old_node_size)); in Extend()
197 TF_RETURN_IF_ERROR(graph::ValidateGraphDef(gdef, *flib_def_)); in Extend()
213 &new_execution_state->original_graph_def_, *flib_def_, 0)); in Extend()
568 optimization_options.flib_def = flib_def_.get(); in InitBaseGraph()
615 if (flib_def_) { in OptimizeGraph()
[all …]
|
| D | partitioning_utils.h | 76 : flib_def_(flib_def), name_(name), counter_(0) {} in FunctionNameGenerator()
83 const FunctionLibraryDefinition* flib_def_;
|
| D | graph_execution_state.h | 205 std::unique_ptr<FunctionLibraryDefinition> flib_def_; variable
|
| D | partitioning_utils.cc | 137 if (flib_def_->Find(candidate) == nullptr) { in GetName()
|
| D | direct_session.cc | 354 flib_def_.reset(nullptr); in ~DirectSession()
360 if (flib_def_ && execution_state_) { in MaybeInitializeExecutionState()
367 flib_def_.reset( in MaybeInitializeExecutionState()
415 TF_RETURN_IF_ERROR(flib_def_->AddLibrary(graph.library())); in ExtendLocked()
1528 run_state_args->graph.reset(new Graph(flib_def_.get())); in CreateGraphs()
|
| D | direct_session.h | 383 std::unique_ptr<FunctionLibraryDefinition> flib_def_; variable
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | functionalize_cond_test.cc | 34 flib_def_.reset( in FunctionalizeCondTest()
37 flib_def_.get())); in FunctionalizeCondTest()
61 std::unique_ptr<FunctionLibraryDefinition> flib_def_; member in tensorflow::functionalize_cond::FunctionalizeCondTest
|
| D | xla_compiler_test.cc | 60 flib_def_.reset(new FunctionLibraryDefinition(OpRegistry::Global(), flib)); in SetUp()
67 options.flib_def = flib_def_.get(); in DefaultOptions()
76 std::unique_ptr<FunctionLibraryDefinition> flib_def_; member in tensorflow::XlaCompilerTest
899 TF_ASSERT_OK(flib_def_->AddFunctionDef(FillFn())); in TEST_F()
909 TF_ASSERT_OK(NodeDefBuilder("fill", "FillFn", flib_def_.get()) in TEST_F()
1342 TF_ASSERT_OK(flib_def_->AddFunctionDef(fn)); in TEST_F()
1352 TF_ASSERT_OK(NodeDefBuilder("fill_fn", "FillFn", flib_def_.get()) in TEST_F()
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | kernel_and_device_test.cc | 43 TestEnv() : flib_def_(OpRegistry::Global(), {}) { in TestEnv()
51 device_mgr_.get(), Env::Default(), TF_GRAPH_DEF_VERSION, &flib_def_,
63 FunctionLibraryDefinition flib_def_; member in tensorflow::__anon9d2e75500111::TestEnv
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | graph_rewrite_dataset.cc | 55 TF_RETURN_IF_ERROR(ctx->function_library()->Clone(&flib_def_, &pflr_, &lib_)); in Optimize()
64 AddToFunctionLibrary(flib_def_.get(), graph_def.library())); in Optimize()
88 params.flib_def = flib_def_.get(); in AsGraphDefInternal()
|
| D | graph_rewrite_dataset.h | 85 std::unique_ptr<FunctionLibraryDefinition> flib_def_ = nullptr; variable
|
| D | multi_device_iterator_ops.cc | 57 flib_def_(std::move(flib_def)), in MultiDeviceIterator()
357 const std::unique_ptr<FunctionLibraryDefinition> flib_def_; member in tensorflow::data::__anond56f2ce10111::MultiDeviceIterator
|
| /external/tensorflow/tensorflow/core/common_runtime/data/ |
| D | standalone.h | 111 std::unique_ptr<FunctionLibraryDefinition> flib_def_; variable
|
| D | standalone.cc | 116 flib_def_(flib_def), in Dataset()
|