| /aosp_15_r20/external/tensorflow/tensorflow/core/framework/ |
| H A D | function_test.cc | 71 auto fdef = FDH::Create( in TEST() local
118 auto fdef = FDH::Create( in TEST() local
176 auto fdef = FDH::Create( in TEST() local
222 auto fdef = FDH::Create( in TEST() local
271 auto fdef = FDH::Create( in TEST() local
308 auto fdef = FDH::Create( in TEST() local
362 auto fdef = FDH::Create( in TEST() local
409 auto fdef = FDH::Define( in TEST() local
478 auto fdef = FDH::Create( in TEST() local
551 auto fdef = FDH::Define( in TEST() local
[all …]
|
| H A D | graph_to_functiondef.cc | 160 FunctionDef* fdef) { in FillFunctionBody()
298 const std::vector<string>& output_names, FunctionDef* fdef) { in GraphToFunctionDefHelper()
377 const char* description, FunctionDef* fdef) { in GraphToFunctionDef()
589 FunctionDef* fdef) { in GraphToFunctionDef()
595 FunctionDef* fdef) { in GraphToFunctionDef()
601 FunctionDef* fdef) { in GraphToFunctionDef()
|
| H A D | function.cc | 597 string Print(const FunctionDef& fdef) { in Print()
737 Status InstantiateFunction(const FunctionDef& fdef, AttrSlice attr_values, in InstantiateFunction()
874 for (const auto& fdef : gdef.library().function()) { in DebugStringWhole() local
889 std::map<string, AttrValue> GetSetAttrs(const FunctionDef& fdef) { in GetSetAttrs()
930 uint64 FunctionDefHash(const FunctionDef& fdef) { in FunctionDefHash()
1241 for (const auto& fdef : def_lib.function()) { in FunctionLibraryDefinition() local
1279 const FunctionDef& fdef, const StackTracesMap& stack_traces) { in AddFunctionDef()
1286 const FunctionDef& fdef, const StackTracesMap& stack_traces, bool* added) { in AddFunctionDefHelper()
1453 for (const FunctionDef& fdef : lib_def.function()) { in AddLibrary() local
1483 const string& func, const FunctionDef& fdef, in ReplaceFunction()
[all …]
|
| H A D | graph_to_functiondef_test.cc | 72 FunctionDef fdef; in TEST() local
109 FunctionDef fdef; in TEST() local
148 FunctionDef fdef; in TEST() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/cc/experimental/libtf/tests/ |
| H A D | function_test.cc | 153 FunctionDef fdef = SquareFunc(); in TEST_P() local
171 FunctionDef fdef = AddFunc(); in TEST_P() local
194 FunctionDef fdef = IdentityNFunc(); in TEST_P() local
217 FunctionDef fdef = SquareFunc(); in TEST_P() local
239 FunctionDef fdef = IdentityNFunc(); in TEST_P() local
264 FunctionDef fdef = AddFunc(); in TEST_P() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/c/ |
| H A D | c_api_function_test.cc | 229 FunctionDef fdef() { in fdef() function in tensorflow::__anon10e0d3b10111::CApiFunctionTest
230 tensorflow::FunctionDef fdef; in fdef() local
252 void VerifyFDefNodes(const tensorflow::FunctionDef& fdef, in VerifyFDefNodes()
265 void VerifyFDefInputs(const tensorflow::FunctionDef& fdef, in VerifyFDefInputs()
282 void VerifyFDefOutputs(const tensorflow::FunctionDef& fdef, in VerifyFDefOutputs()
300 const tensorflow::FunctionDef& fdef, in VerifyFDefEdges()
359 tensorflow::FunctionDef fdef; in VerifyFDef() local
370 tensorflow::FunctionDef fdef; in Reincarnate() local
952 tensorflow::FunctionDef fdef; in TEST_F() local
1546 tensorflow::FunctionDef fdef; in TEST_F() local
[all …]
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/ |
| H A D | xla_kernel_creator_test.cc | 70 for (const auto& fdef : flib) { in Init() local
99 FunctionDef fdef = XTimesY(); in TEST_F() local
128 FunctionDef fdef = XTimesY(); in TEST_F() local
144 FunctionDef fdef = XTimesY(); in TEST_F() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/utils/ |
| H A D | tfrt_graph_execution_state_test.cc | 342 auto fdef = tensorflow::FunctionDefHelper::Create( in TEST_F() local
399 auto fdef = tensorflow::FunctionDefHelper::Create( in TEST_F() local
584 auto fdef = tensorflow::FunctionDefHelper::Create( in TEST_F() local
651 auto fdef = tensorflow::FunctionDefHelper::Create( in TEST_F() local
803 for (const FunctionDef& fdef : graphdef.library().function()) { in GetGraphInfo() local
809 const FunctionDef& fdef = in GetGraphInfo() local
882 for (const FunctionDef& fdef : graph_info.fdefs) { in TEST_F() local
907 auto fdef = tensorflow::FunctionDefHelper::Create( in TEST_F() local
993 for (const FunctionDef& fdef : graph_info.fdefs) { in TEST_F() local
|
| H A D | graph_partition_test.cc | 178 for (const FunctionDef& fdef : new_graphdef.library().function()) { in TEST_F() local
190 const FunctionDef& fdef = in TEST_F() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/framework/ |
| H A D | function_def_to_graph.py | 34 def function_def_to_graph(fdef, argument
149 def function_def_to_graph_def(fdef, input_shapes=None): argument
304 def _set_handle_data(func_graph, fdef): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/data/ |
| H A D | captured_function.cc | 232 const FunctionDef* fdef = lib_def->Find(func_name); in CreateFunctionLibraryDefinition() local
243 const string& name, const FunctionDef** fdef) { in LookupFunction()
447 const FunctionDef* fdef; in Create() local
573 const FunctionDef* fdef; in Instantiate() local
703 const FunctionDef* fdef; in IsMultiDevice() local
739 const FunctionDef* fdef; in IsMultiDevice() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/c/eager/ |
| H A D | graph_function.cc | 20 GraphFunction::GraphFunction(FunctionDef fdef) in GraphFunction()
23 Status GraphFunction::GetFunctionDef(FunctionDef** fdef) { in GetFunctionDef()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/ |
| H A D | function_def_utils.cc | 30 const FunctionDef& fdef, const AttrSlice& attrs, in FunctionDefToBodyHelper()
76 Status FunctionDefToBodyHelper(const FunctionDef& fdef, const AttrSlice& attrs, in FunctionDefToBodyHelper()
|
| H A D | placer_inspection_required_ops_utils.cc | 66 const FunctionDef* fdef; in IsPlacerInspectionRequired() local
81 const Node& node, const FunctionDef** fdef, in GetFunctionDefAndAttrs()
|
| H A D | function.cc | 678 const FunctionDef& fdef, AttrSlice attrs, in FunctionDefToBody()
694 const FunctionDef* fdef = lib_def->Find(func.name()); in InstantiateSymbolicGradient() local
804 const FunctionDef* fdef = lib_def->Find(function_name); in Instantiate() local
892 void PruneFunctionBody(const FunctionDef& fdef, Graph* g) { in PruneFunctionBody()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/data/ |
| H A D | noop_elimination.cc | 80 bool IsOutputIdentityOfInput(const FunctionDef& fdef, const string& output_arg, in IsOutputIdentityOfInput()
122 const FunctionDef* fdef = in IsMapIdentity() local
|
| H A D | function_utils.cc | 91 FunctionDef* fdef, DataType dtype) { in AddFunctionOutputWithUniqueName()
105 OpDef_ArgDef* AddFunctionInput(const string& name, FunctionDef* fdef, in AddFunctionInput()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/flex/ |
| H A D | delegate_data.cc | 57 tensorflow::FunctionDef& fdef) { in BuildFunctionDefProto()
185 tensorflow::FunctionDef fdef; in RegisterFunctionDefForSubgraphs() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/ |
| H A D | function_optimizer.cc | 148 bool MarkedNoSpecialize(const FunctionDef& fdef) { in MarkedNoSpecialize()
964 const FunctionDef* fdef = flib_def.Find(n->type_string()); in ValidateNoDeadOutputs() local
1021 const FunctionDef** fdef) -> Status { in MakeFunctionBodyForInlining()
1066 const FunctionDef* fdef; in MakeFunctionBodyForInlining() local
1079 const FunctionDef* fdef; in MakeFunctionBodyForInlining() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/ |
| H A D | tf2xla_util_test.cc | 435 FunctionDef fdef; in TEST() local
449 FunctionDef fdef; in TEST() local
464 FunctionDef fdef; in TEST() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/distributed_runtime/integration_test/ |
| H A D | c_api_multi_client_function_test.cc | 255 const std::string& fdef = in TEST_P() local
298 const std::string& fdef = in TEST_P() local
|
| H A D | c_api_coordination_test.cc | 61 FunctionDef fdef; in SetConfigKeyValueFn() local
87 FunctionDef fdef; in GetConfigKeyValueFn() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/saved_model/ |
| H A D | function_deserialization.py | 578 def _fix_fdef_in_place(fdef, functions, shared_name_suffix, argument
619 def _list_function_deps(fdef, library_function_names, library_gradient_names): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/eager/ |
| H A D | context.cc | 836 Status EagerContext::MaybeRegisterFunctionRemotely(const FunctionDef& fdef) { in MaybeRegisterFunctionRemotely()
919 const FunctionDef& fdef, const StackTracesMap& stack_traces) { in AddFunctionDefWithStackTraces()
924 Status EagerContext::AddFunctionDef(const FunctionDef& fdef) { in AddFunctionDef()
929 Status EagerContext::AddFunctionDef(const FunctionDef& fdef, in AddFunctionDef()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/eager/ |
| H A D | c_api_tfrt.cc | 914 tensorflow::FunctionDef* fdef; in RegisterFunction() local
1090 const tensorflow::FunctionDef& fdef) { in AddFunctionDef()
1095 const tensorflow::FunctionDef& fdef, in AddFunctionDefWithStackTraces()
1210 const tensorflow::FunctionDef* fdef = in RunMetadataRecordFunction() local
|