Home
last modified time | relevance | path

Searched refs:FunctionDefLibrary (Results 1 – 25 of 116) sorted by relevance

12345

/aosp_15_r20/out/soong/.intermediates/external/tensorflow/tensorflow_core_proto_cpp_lite/android_arm64_armv8-2a_cortex-a55_static_apex33/gen/proto/tensorflow/core/framework/
Dfunction.pb.h55 class FunctionDefLibrary; variable
88 template<> ::tensorflow::FunctionDefLibrary* Arena::CreateMaybeMessage<::tensorflow::FunctionDefLib…
103 class FunctionDefLibrary final :
106 inline FunctionDefLibrary() : FunctionDefLibrary(nullptr) {} in FunctionDefLibrary() function
107 ~FunctionDefLibrary() override;
108 …explicit PROTOBUF_CONSTEXPR FunctionDefLibrary(::PROTOBUF_NAMESPACE_ID::internal::ConstantInitiali…
110 FunctionDefLibrary(const FunctionDefLibrary& from);
111 FunctionDefLibrary(FunctionDefLibrary&& from) noexcept in FunctionDefLibrary() function
112 : FunctionDefLibrary() { in FunctionDefLibrary()
116 inline FunctionDefLibrary& operator=(const FunctionDefLibrary& from) {
[all …]
Dgraph.pb.h196 const ::tensorflow::FunctionDefLibrary& library() const; in Swap()
197 PROTOBUF_NODISCARD ::tensorflow::FunctionDefLibrary* release_library(); in Swap()
198 ::tensorflow::FunctionDefLibrary* mutable_library(); in Swap()
199 void set_allocated_library(::tensorflow::FunctionDefLibrary* library); in Swap()
201 const ::tensorflow::FunctionDefLibrary& _internal_library() const; in Swap()
202 ::tensorflow::FunctionDefLibrary* _internal_mutable_library(); in Swap()
205 ::tensorflow::FunctionDefLibrary* library); in Swap()
206 ::tensorflow::FunctionDefLibrary* unsafe_arena_release_library(); in Swap()
244 ::tensorflow::FunctionDefLibrary* library_; in Swap()
407 inline const ::tensorflow::FunctionDefLibrary& GraphDef::_internal_library() const { in _internal_library()
[all …]
Dfunction.pb.cc22 PROTOBUF_CONSTEXPR FunctionDefLibrary::FunctionDefLibrary( in FunctionDefLibrary() function in tensorflow::FunctionDefLibrary
33 FunctionDefLibrary _instance;
168 class FunctionDefLibrary::_Internal {
172 FunctionDefLibrary::FunctionDefLibrary(::PROTOBUF_NAMESPACE_ID::Arena* arena, in FunctionDefLibrary() function in tensorflow::FunctionDefLibrary
178 FunctionDefLibrary::FunctionDefLibrary(const FunctionDefLibrary& from) in FunctionDefLibrary() function in tensorflow::FunctionDefLibrary
180 FunctionDefLibrary* const _this = this; (void)_this; in FunctionDefLibrary()
191 inline void FunctionDefLibrary::SharedCtor( in SharedCtor()
203 FunctionDefLibrary::~FunctionDefLibrary() { in ~FunctionDefLibrary()
212 inline void FunctionDefLibrary::SharedDtor() { in SharedDtor()
219 void FunctionDefLibrary::SetCachedSize(int size) const { in SetCachedSize()
[all …]
Dgraph.pb.cc46 static const ::tensorflow::FunctionDefLibrary& library(const GraphDef* msg);
53 const ::tensorflow::FunctionDefLibrary&
90 _this->_impl_.library_ = new ::tensorflow::FunctionDefLibrary(*from._impl_.library_); in GraphDef()
320 _this->_internal_mutable_library()->::tensorflow::FunctionDefLibrary::MergeFrom( in MergeFrom()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/
H A Dbuild_xla_ops_pass_test.cc58 Status BuildXlaOps(const Scope& s, const FunctionDefLibrary& fdef_lib, in BuildXlaOps()
121 FunctionDefLibrary CreateFunctionDefLibWithConstFunction(const string& name) { in CreateFunctionDefLibWithConstFunction()
122 FunctionDefLibrary fdef_lib; in CreateFunctionDefLibWithConstFunction()
136 FunctionDefLibrary fdef_lib = in TEST_F()
158 FunctionDefLibrary fdef_lib = in TEST_F()
178 FunctionDefLibrary fdef_lib = in TEST_F()
214 FunctionDefLibrary fdef_lib = in TEST_F()
243 FunctionDefLibrary fdef_lib = in TEST_F()
261 FunctionDefLibrary CreateFunctionDefLibWithInt32Input(const string& name) { in CreateFunctionDefLibWithInt32Input()
262 FunctionDefLibrary fdef_lib; in CreateFunctionDefLibWithInt32Input()
[all …]
H A Dencapsulate_subgraphs_pass_test.cc50 FunctionDefLibrary* library) { in AddGraphDefToFunctionLibrary()
259 bool EqualFunctionDefLibrary(const FunctionDefLibrary& expected, in EqualFunctionDefLibrary()
260 const FunctionDefLibrary& actual, string* diff) { in EqualFunctionDefLibrary()
480 Status Encapsulate(GraphDef* graphdef, FunctionDefLibrary* library, in Encapsulate()
553 Status Encapsulate(GraphDef* graphdef, FunctionDefLibrary* library) { in Encapsulate()
568 FunctionDefLibrary library_in; in TEST()
573 FunctionDefLibrary library_out = library_in; in TEST()
583 FunctionDefLibrary library; in TEST()
603 FunctionDefLibrary library_expected; in TEST()
636 FunctionDefLibrary library; in TEST()
[all …]
H A Dcompilability_check_util_test.cc108 std::unique_ptr<FunctionDefLibrary> func_library_ =
109 std::make_unique<FunctionDefLibrary>();
180 FunctionDefLibrary flib; in TEST_F()
221 FunctionDefLibrary flib; in TEST_F()
301 FunctionDefLibrary flib; in TEST_F()
402 FunctionDefLibrary flib; in TEST_F()
455 FunctionDefLibrary library; in TEST_F()
483 FunctionDefLibrary library; in TEST_F()
511 FunctionDefLibrary library; in TEST_F()
H A Dresource_operation_safety_analysis_test.cc241 FunctionDefLibrary CreateFunctionDefLibWithConstFunction(const string& name) { in CreateFunctionDefLibWithConstFunction()
242 FunctionDefLibrary flib_def; in CreateFunctionDefLibWithConstFunction()
263 FunctionDefLibrary flib_def = in TEST()
285 FunctionDefLibrary flib_def = in TEST()
305 FunctionDefLibrary flib_def = in TEST()
325 FunctionDefLibrary flib_def = in TEST()
347 FunctionDefLibrary flib_def = in TEST()
374 FunctionDefLibrary flib_def = in TEST()
H A Dxla_cluster_util_test.cc140 FunctionDefLibrary flib; in GetNodesRelatedToRefVarsSorted()
206 void AddRefFunctionFunctionDef(FunctionDefLibrary* fdef_lib) { in AddRefFunctionFunctionDef()
217 void AddRegularFunctionFunctionDef(FunctionDefLibrary* fdef_lib) { in AddRegularFunctionFunctionDef()
229 FunctionDefLibrary fdef_lib; in TEST()
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/utils/
H A Dfunctions_test.cc128 FunctionLibraryDefinition flib(OpRegistry::Global(), FunctionDefLibrary()); in TEST_F()
209 FunctionLibraryDefinition flib(OpRegistry::Global(), FunctionDefLibrary()); in TEST_F()
268 FunctionLibraryDefinition flib(OpRegistry::Global(), FunctionDefLibrary()); in TEST_F()
369 FunctionLibraryDefinition flib(OpRegistry::Global(), FunctionDefLibrary()); in TEST_F()
421 FunctionLibraryDefinition flib(OpRegistry::Global(), FunctionDefLibrary()); in TEST_F()
460 FunctionLibraryDefinition flib(OpRegistry::Global(), FunctionDefLibrary()); in TEST_F()
487 FunctionLibraryDefinition flib(OpRegistry::Global(), FunctionDefLibrary()); in TEST_F()
536 FunctionLibraryDefinition flib(OpRegistry::Global(), FunctionDefLibrary()); in TEST_F()
585 FunctionLibraryDefinition flib(OpRegistry::Global(), FunctionDefLibrary()); in TEST_F()
676 FunctionDefLibrary lib_def; in TEST_F()
[all …]
H A Dtpu_test.cc33 FunctionDefLibrary* library = tpu_graph.mutable_library(); in TEST_F()
54 FunctionDefLibrary* library = tpu_graph.mutable_library(); in TEST_F()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/
H A Dresource_util_test.cc86 FunctionDefLibrary()); in TEST()
121 FunctionDefLibrary()); in TEST()
161 FunctionDefLibrary()); in TEST()
204 FunctionDefLibrary()); in TEST()
253 FunctionDefLibrary()); in TEST()
316 FunctionDefLibrary()); in TEST()
378 auto library = std::make_unique<FunctionDefLibrary>(); in TEST()
432 auto library = std::make_unique<FunctionDefLibrary>(); in TEST()
485 auto library = std::make_unique<FunctionDefLibrary>(); in TEST()
H A Dfunctionalize_cond_test.cc65 FunctionDefLibrary fdef_lib_;
143 FunctionDefLibrary fdef_lib; in TEST()
149 FunctionDefLibrary flib_def_proto = flib_def.ToProto(); in TEST()
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/
H A Dfunction_api_info_test.cc71 FunctionDefLibrary* func_lib) { in PopulateSampleLibrary()
89 void PopulateComplexLibrary(FunctionDefLibrary* func_lib) { in PopulateComplexLibrary()
134 FunctionDefLibrary func_lib; in TEST()
156 FunctionDefLibrary func_lib; in TEST()
181 FunctionDefLibrary func_lib; in TEST()
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/data/
H A Dfusion_utils.h60 FunctionDef* fused_function, FunctionDefLibrary* library)>;
64 FunctionDefLibrary* library);
117 FunctionDefLibrary* library);
129 const SetNodesFn& set_nodes, FunctionDefLibrary* library);
H A Dgraph_utils.h106 const FunctionDefLibrary& library);
118 const FunctionDefLibrary& library);
147 const FunctionDefLibrary* library,
/aosp_15_r20/external/tensorflow/tensorflow/core/data/
H A Dhash_utils_test.cc38 uint64 GetHash(const FunctionDefLibrary& library, const FunctionDef& fn) { in GetHash()
52 Status CheckEqual(const FunctionDefLibrary& library, const FunctionDef& fn1, in CheckEqual()
89 FunctionDefLibrary fl; in TEST_F()
112 FunctionDefLibrary fl; in TEST_F()
138 FunctionDefLibrary fl; in TEST_F()
481 FunctionDefLibrary* fl1 = gd.mutable_library(); in TEST_F()
545 FunctionDefLibrary* fl1 = gd.mutable_library(); in TEST_F()
612 FunctionDefLibrary* fl1 = gd.mutable_library(); in TEST_F()
660 FunctionDefLibrary* fl1 = gd.mutable_library(); in TEST_F()
712 FunctionDefLibrary* fl1 = gd.mutable_library(); in TEST_F()
[all …]
H A Ddataset_utils_test.cc73 FunctionDefLibrary fdef_base; in TEST()
78 FunctionDefLibrary fdef_to_add; in TEST()
99 FunctionDefLibrary fdef_base; in TEST()
108 FunctionDefLibrary fdef_to_add; in TEST()
136 FunctionDefLibrary flib; in TEST()
/aosp_15_r20/external/tensorflow/tensorflow/core/framework/
H A Dfunction_test.cc1103 FunctionLibraryDefinition lib_def(OpRegistry::Global(), FunctionDefLibrary()); in TEST()
1150 FunctionDefLibrary proto; in TEST()
1205 FunctionDefLibrary proto; in TEST()
1212 FunctionLibraryDefinition lib_def(OpRegistry::Global(), FunctionDefLibrary()); in TEST()
1248 FunctionDefLibrary proto; in TEST()
1257 FunctionDefLibrary proto2; in TEST()
1284 FunctionDefLibrary proto; in TEST()
1293 FunctionDefLibrary proto2; in TEST()
1318 FunctionDefLibrary proto = lib_def1.ToProto(); in TEST()
1344 FunctionDefLibrary proto; in TEST()
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/utils/
H A Dtfrt_graph_execution_state_test.cc336 tensorflow::FunctionDefLibrary fdef_lib; in TEST_F()
348 tensorflow::FunctionDefLibrary fdef_lib; in TEST_F()
415 tensorflow::FunctionDefLibrary fdef_lib; in TEST_F()
441 tensorflow::FunctionDefLibrary fdef_lib; in TEST_F()
467 tensorflow::FunctionDefLibrary fdef_lib; in TEST_F()
549 tensorflow::FunctionDefLibrary fdef_lib; in TEST_F()
578 tensorflow::FunctionDefLibrary fdef_lib; in TEST_F()
590 tensorflow::FunctionDefLibrary fdef_lib; in TEST_F()
645 tensorflow::FunctionDefLibrary fdef_lib; in TEST_F()
657 tensorflow::FunctionDefLibrary fdef_lib; in TEST_F()
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/
H A Dexport_graphdef.cc124 FunctionDefLibrary* flib, llvm::SmallDenseSet<FuncOp>& visited_functions);
133 FunctionDefLibrary* flib, llvm::SmallDenseSet<FuncOp>& visited_functions,
425 const SymbolTable& symbol_table, FuncOp function, FunctionDefLibrary* flib, in Convert()
590 const SymbolTable& symbol_table, FuncOp function, FunctionDefLibrary* flib, in ConvertLibFunction()
673 FunctionDefLibrary flib; in Convert()
743 FunctionDefLibrary()); in ConvertMlirToGraphdef()
752 FunctionDefLibrary flib = flib_def.ToProto(); in ConvertMlirToGraphdef()
775 FunctionDefLibrary flib; in ConvertMlirFunctionToFunctionLibraryDef()
/aosp_15_r20/external/tensorflow/tensorflow/python/framework/
H A Dgraph_util_test.py399 library = function_pb2.FunctionDefLibrary()
422 library = function_pb2.FunctionDefLibrary()
428 reversed_library = function_pb2.FunctionDefLibrary()
443 library = function_pb2.FunctionDefLibrary()
454 reversed_library = function_pb2.FunctionDefLibrary()
H A Dproto_comparators.cc59 FunctionDefLibrary::descriptor()->FindFieldByName("function")); in EqualsGraphDef()
61 FunctionDefLibrary::descriptor()->FindFieldByName("gradient")); in EqualsGraphDef()
/aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/fallback/
H A Dfallback_state.h39 const tensorflow::FunctionDefLibrary &fdef_lib);
43 const tensorflow::FunctionDefLibrary &fdef_lib);
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/
H A Dlower_while_op_test.cc61 FunctionDefLibrary f_lib_proto; in TEST()
178 FunctionDefLibrary f_lib_proto; in TEST()
271 FunctionDefLibrary f_lib_proto; in TEST()
410 FunctionDefLibrary f_lib_proto; in TEST()
526 FunctionDefLibrary f_lib_proto; in TEST()

12345