| /external/tensorflow/tensorflow/core/common_runtime/ |
| process_function_library_runtime.cc | 30 const FunctionLibraryDefinition* lib_def,
52 const FunctionLibraryDefinition* lib_def,
339 std::unique_ptr<FunctionLibraryDefinition>* out_lib_def,
341 out_lib_def->reset(new FunctionLibraryDefinition(*lib_def_));
|
| graph_execution_state.cc | 57 flib_def_(new FunctionLibraryDefinition(OpRegistry::Global(),
418 std::unique_ptr<FunctionLibraryDefinition> flib(
419 new FunctionLibraryDefinition(*flib_def_));
|
| shape_refiner_test.cc | [all...] |
| function_test.cc | 147 lib_def_.reset(new FunctionLibraryDefinition(OpRegistry::Global(), proto));
342 std::unique_ptr<FunctionLibraryDefinition> lib_def_;
388 std::unique_ptr<FunctionLibraryDefinition> overlay_lib(
389 new FunctionLibraryDefinition(OpRegistry::Global(), proto));
[all...] |
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| graph_mgr.h | 130 std::unique_ptr<FunctionLibraryDefinition> lib_def;
|
| cluster_function_library_runtime.cc | 121 const string& function_name, const FunctionLibraryDefinition& lib_def,
|
| graph_mgr.cc | 125 new FunctionLibraryDefinition(OpRegistry::Global(), gdef.library()));
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| graph_compiler.cc | 168 return n->type_string() == FunctionLibraryDefinition::kGradientOp ||
186 func.set_name(FunctionLibraryDefinition::kGradientOp);
|
| functionalize_control_flow_test.cc | 95 FunctionLibraryDefinition library(OpRegistry::Global(), {});
238 FunctionLibraryDefinition library(OpRegistry::Global(), {});
341 FunctionLibraryDefinition library(OpRegistry::Global(), {});
482 FunctionLibraryDefinition library(OpRegistry::Global(), {});
715 FunctionLibraryDefinition library(OpRegistry::Global(), {});
[all...] |
| functionalize_control_flow.cc | 286 FunctionLibraryDefinition* library) {
580 static Status Functionalize(Graph* graph, FunctionLibraryDefinition* library);
620 FunctionalizeCond(Graph* graph, FunctionLibraryDefinition* library,
685 FunctionLibraryDefinition* library_;
[all...] |
| xla_compiler_test.cc | 53 flib_def_.reset(new FunctionLibraryDefinition(OpRegistry::Global(), flib));
64 FunctionLibraryDefinition* LocalFlibDef(XlaCompiler* compiler) {
70 std::unique_ptr<FunctionLibraryDefinition> flib_def_;
|
| tf2xla.cc | 319 FunctionLibraryDefinition flib_def(OpRegistry::Global(), graph_def.library());
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| iterator_ops.cc | 88 std::unique_ptr<FunctionLibraryDefinition> flib_def,
145 std::shared_ptr<FunctionLibraryDefinition> flib_def(
146 new FunctionLibraryDefinition(
181 std::shared_ptr<const FunctionLibraryDefinition> function_library() {
221 std::unique_ptr<FunctionLibraryDefinition> flib_def_;
227 std::shared_ptr<const FunctionLibraryDefinition> lib_def_ GUARDED_BY(mu_);
468 std::unique_ptr<FunctionLibraryDefinition> flib_def(nullptr);
543 std::unique_ptr<FunctionLibraryDefinition>* flib_def,
553 flib_def->reset(new FunctionLibraryDefinition(
739 std::unique_ptr<FunctionLibraryDefinition> flib_def(nullptr)
[all...] |
| /external/tensorflow/tensorflow/compiler/jit/ |
| encapsulate_subgraphs_pass.cc | 171 FunctionLibraryDefinition* library);
176 FunctionLibraryDefinition* library);
246 FunctionLibraryDefinition* library);
338 Status ReplaceFunctionDef(FunctionLibraryDefinition* library);
552 FunctionLibraryDefinition* library,
564 FunctionLibraryDefinition* library);
575 FunctionLibraryDefinition* library);
584 Graph* graph_out, FunctionLibraryDefinition* library);
[all...] |
| mark_for_compilation_pass_test.cc | 39 FunctionLibraryDefinition* flib_def) {
55 FunctionLibraryDefinition flib_def((*graph)->op_registry(), flib);
204 FunctionLibraryDefinition flib_def(OpRegistry::Global(), flib);
|
| mark_for_compilation_pass.cc | 182 const Graph& graph, FunctionLibraryDefinition* flib_def, Env* env,
318 const FunctionLibraryDefinition* fld = options.flib_def;
|
| /external/tensorflow/tensorflow/core/framework/ |
| dataset.h | 156 // name `function_name` is not found in the FunctionLibraryDefinition, returns
171 const FunctionLibraryDefinition* lib_def =
274 std::shared_ptr<const FunctionLibraryDefinition> function_library = nullptr;
296 std::shared_ptr<const FunctionLibraryDefinition> function_library() {
|
| dataset.cc | 144 const FunctionLibraryDefinition* flib_def =
|
| /external/tensorflow/tensorflow/core/graph/ |
| graph.h | 407 explicit Graph(const FunctionLibraryDefinition& flib_def);
543 const FunctionLibraryDefinition& flib_def() const { return ops_; }
599 FunctionLibraryDefinition ops_;
|
| /external/tensorflow/tensorflow/core/grappler/ |
| grappler_item_builder.cc | 103 FunctionLibraryDefinition function_library(OpRegistry::Global(),
455 FunctionLibraryDefinition(OpRegistry::Global(),
547 FunctionLibraryDefinition func_def(OpRegistry::Global(), library);
|
| utils.cc | 227 FunctionLibraryDefinition fdef(OpRegistry::Global(), graph->library());
|
| grappler_item_builder_test.cc | 90 if (node.op() == FunctionLibraryDefinition::kGradientOp) {
|
| /external/tensorflow/tensorflow/c/eager/ |
| runtime_test.cc | 52 FunctionLibraryDefinition flib_def_;
|
| /external/tensorflow/tensorflow/contrib/tensorrt/segment/ |
| segment.cc | 124 tensorflow::FunctionLibraryDefinition flib(tensorflow::OpRegistry::Global(),
|
| /external/tensorflow/tensorflow/contrib/tensorrt/convert/ |
| convert_graph.cc | 231 tensorflow::FunctionLibraryDefinition flib(tensorflow::OpRegistry::Global(),
|
