| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| functionalize_control_flow.h | 27 FunctionLibraryDefinition* library);
|
| dump_graph.h | 41 const FunctionLibraryDefinition* flib_def = nullptr);
|
| test_util.h | 42 const FunctionLibraryDefinition& library,
|
| test_util.cc | 24 const FunctionLibraryDefinition& library,
|
| xla_compiler.h | 236 const FunctionLibraryDefinition* flib_def = nullptr;
344 std::unique_ptr<FunctionLibraryDefinition> local_flib_def_;
|
| dump_graph.cc | 76 const FunctionLibraryDefinition* flib_def) {
|
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| add_default_attributes.cc | 28 std::unique_ptr<FunctionLibraryDefinition> flib_def(
29 new FunctionLibraryDefinition(OpRegistry::Global(),
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| function.h | 57 int graph_def_version, const FunctionLibraryDefinition* lib_def,
67 int graph_def_version, const FunctionLibraryDefinition* lib_def,
159 void InlineFunctionBody(const FunctionLibraryDefinition& flib_def, Graph* g,
166 const FunctionLibraryDefinition* const lib_def,
|
| optimization_registry_test.cc | 62 new FunctionLibraryDefinition(OpRegistry::Global(), func_def_lib));
78 std::unique_ptr<FunctionLibraryDefinition> flib_def_;
|
| graph_execution_state.h | 52 explicit ClientGraph(std::unique_ptr<FunctionLibraryDefinition> flib,
60 std::unique_ptr<FunctionLibraryDefinition> flib_def;
192 std::unique_ptr<FunctionLibraryDefinition> flib_def_;
|
| process_function_library_runtime.h | 34 const FunctionLibraryDefinition* lib_def,
41 const FunctionLibraryDefinition* lib_def,
141 std::unique_ptr<FunctionLibraryDefinition>* out_lib_def,
159 const FunctionLibraryDefinition* lib_def_;
|
| shape_refiner.h | 164 const tensorflow::FunctionLibraryDefinition* lib) {
290 const tensorflow::FunctionLibraryDefinition* function_library_ = nullptr;
|
| function.cc | 51 FunctionLibraryDefinition::kGradientOp;
54 FunctionLibraryDefinition::kFuncAttr;
146 const FunctionLibraryDefinition* lib_def,
174 const FunctionLibraryDefinition* GetFunctionLibraryDefinition()
185 Status Clone(std::unique_ptr<FunctionLibraryDefinition>* out_lib_def,
196 const FunctionLibraryDefinition* const base_lib_def_;
212 const FunctionLibraryDefinition* overlay_lib = nullptr; // Not owned.
226 const FunctionLibraryDefinition* lib_def,
229 const FunctionLibraryDefinition* lib_def,
234 const FunctionLibraryDefinition* lib_def
[all...] |
| optimization_registry.h | 42 FunctionLibraryDefinition* flib_def = nullptr; // Not owned.
|
| process_function_library_runtime_test.cc | 43 lib_def_.reset(new FunctionLibraryDefinition(OpRegistry::Global(), proto));
108 std::unique_ptr<FunctionLibraryDefinition> lib_def_;
115 std::unique_ptr<FunctionLibraryDefinition> lib_def(
116 new FunctionLibraryDefinition(OpRegistry::Global(), proto));
|
| direct_session.h | 156 std::unique_ptr<FunctionLibraryDefinition> flib_def;
215 std::unique_ptr<FunctionLibraryDefinition>* flib_def,
330 std::unique_ptr<FunctionLibraryDefinition> flib_def_;
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| encapsulate_subgraphs_pass.h | 80 std::unique_ptr<Graph>* graph_out, FunctionLibraryDefinition* library);
|
| encapsulate_subgraphs_pass_test.cc | 409 std::unique_ptr<FunctionLibraryDefinition> lib_def(
410 new FunctionLibraryDefinition(OpRegistry::Global(), *library));
490 std::unique_ptr<FunctionLibraryDefinition> lib_def(
491 new FunctionLibraryDefinition(OpRegistry::Global(), library_expected));
550 std::unique_ptr<FunctionLibraryDefinition> lib_def(
551 new FunctionLibraryDefinition(OpRegistry::Global(), library_expected));
612 FunctionLibraryDefinition library(OpRegistry::Global(), {});
644 FunctionLibraryDefinition library(OpRegistry::Global(), {});
702 FunctionLibraryDefinition library(OpRegistry::Global(), {});
747 FunctionLibraryDefinition library(OpRegistry::Global(), {})
[all...] |
| /external/tensorflow/tensorflow/core/framework/ |
| function.h | 287 class FunctionLibraryDefinition : public OpRegistryInterface {
289 explicit FunctionLibraryDefinition(const FunctionLibraryDefinition& lib_def);
290 FunctionLibraryDefinition(const OpRegistryInterface* default_registry,
292 ~FunctionLibraryDefinition() override;
294 FunctionLibraryDefinition& operator=(const FunctionLibraryDefinition&) =
327 Status AddLibrary(const FunctionLibraryDefinition& other);
432 // `FunctionLibraryDefinition` to store an `outer_scope` pointer
434 const FunctionLibraryDefinition* overlay_lib = nullptr
[all...] |
| function_test.cc | 912 FunctionLibraryDefinition lib_def(OpRegistry::Global(), proto);
932 FunctionLibraryDefinition lib_def(OpRegistry::Global(), proto);
953 FunctionLibraryDefinition lib_def(OpRegistry::Global(), proto);
988 FunctionLibraryDefinition lib_def(OpRegistry::Global(), FunctionDefLibrary());
1013 FunctionLibraryDefinition lib_def(OpRegistry::Global(), proto);
1027 FunctionLibraryDefinition lib_def2(OpRegistry::Global(), proto);
[all...] |
| function.cc | 905 FunctionLibraryDefinition::FunctionDefAndOpRegistration::
913 FunctionLibraryDefinition::FunctionLibraryDefinition(
914 const FunctionLibraryDefinition& other)
921 FunctionLibraryDefinition::FunctionLibraryDefinition(
936 FunctionLibraryDefinition::~FunctionLibraryDefinition() {}
938 const FunctionDef* FunctionLibraryDefinition::Find(const string& name) const {
947 Status FunctionLibraryDefinition::AddFunctionDef(const FunctionDef& fdef)
[all...] |
| /external/tensorflow/tensorflow/core/graph/ |
| graph_partition.h | 51 const FunctionLibraryDefinition* flib_def = nullptr;
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| cluster_function_library_runtime.h | 36 const FunctionLibraryDefinition& lib_def, AttrSlice attrs,
|
| cluster_function_library_runtime_test.cc | 59 const FunctionLibraryDefinition& lib_def,
68 const string& function_name, const FunctionLibraryDefinition& lib_def,
310 FunctionLibraryDefinition lib_def(OpRegistry::Global(), proto);
325 FunctionLibraryDefinition lib_def(OpRegistry::Global(), proto);
|
| /external/tensorflow/tensorflow/c/eager/ |
| c_api_internal.h | 68 tensorflow::FunctionLibraryDefinition func_lib_def GUARDED_BY(functions_mu){
|
