Home
last modified time | relevance | path

Searched defs:function_def (Results 1 – 25 of 47) sorted by relevance

12

/aosp_15_r20/external/tensorflow/tensorflow/c/experimental/saved_model/core/
H A Dsaved_model_utils.cc133 const FunctionDef* function_def) { in ValidateSavedFunctionCompatibleWithFunctionDef()
254 const FunctionDef* function_def, in LoadTFConcreteFunction()
377 for (const FunctionDef& function_def : library.function()) { in FunctionNameToFunctionDefMap() local
447 const FunctionDef* function_def = function_def_map.at(function_name); in PartiallyReviveSavedModelObjects() local
476 const FunctionDef* function_def = function_def_map.at(function_name); in PartiallyReviveSavedModelObjects() local
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/data/
H A Dmake_deterministic.cc219 const FunctionDef& function_def, in GetAllTransitiveDependencies()
265 const FunctionDef* function_def = library.Find(func.name()); in SplitMap() local
525 const FunctionDef* function_def = library.Find(function_name); in FunctionMayIntroduceNondeterminism() local
655 const FunctionDef* function_def = library.Find(function_name); in GraphMayHaveAsyncNondeterminism() local
H A Dfunction_utils.cc184 const FunctionDef& function_def, bool skip_assert) { in IsFunctionStateful()
/aosp_15_r20/external/rust/android-crates-io/crates/mls-rs-codec-derive/src/
Dlib.rs309 let function_def = quote! {fn mls_encoded_len(&self) -> usize }; in derive_size() localVariable
320 …let function_def = quote! { fn mls_encode(&self, writer: &mut mls_rs_codec::Vec<u8>) -> Result<(),… in derive_encode() localVariable
331 let function_def = in derive_decode() localVariable
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/utils/
H A Dtpu_test.cc34 FunctionDef* function_def = library->add_function(); in TEST_F() local
55 FunctionDef* function_def = library->add_function(); in TEST_F() local
H A Dtpu.cc31 for (const auto& function_def : def.library().function()) { in IsLegacyTPUBridgeGraphDef() local
/aosp_15_r20/external/tensorflow/tensorflow/c/eager/
H A Dc_api_distributed_test.cc191 const string function_def = AddVariablesFunction(); in TestFunctionWithPackedInput() local
382 const string function_def = VariableAddFunction(); in TEST() local
505 const string function_def = inject_error ? VariableAddFunctionWithGraphError() in TestDistributedFunctionCancellation() local
H A Dc_api_remote_test_util.cc140 string function_def = MatMulFunction(matmul_device); in TestRemoteExecuteSilentCopies() local
H A Dc_api_test.cc1132 string function_def = MatMulFunction(); in FunctionDefAndExecute() local
1182 string function_def = AddFunction(); in RunAddFunction() local
1284 string function_def = MatMulFunction(); in BM_ExecuteFunction() local
H A Dc_api.cc87 tensorflow::FunctionDef& function_def) { in AnnotateEagerRuntimeConstructionContext()
912 tensorflow::FunctionDef function_def; in TFE_ContextAddFunctionDef() local
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/inference/
H A Dbatch_op_rewriter.cc165 FunctionDef* function_def = in Optimize() local
196 FunctionDef* function_def = in Optimize() local
H A Dbatch_op_rewriter_test.cc77 FunctionDef* function_def = function_def_lib->add_function(); local
/aosp_15_r20/external/tensorflow/tensorflow/core/data/
H A Drewrite_utils.cc80 void AddFakeSinks(FunctionDef* function_def) { in AddFakeSinks()
95 void RemoveFakeSinks(FunctionDef* function_def) { in RemoveFakeSinks()
/aosp_15_r20/external/tensorflow/tensorflow/dtensor/cc/
H A Ddtensor_device_util.cc764 const FunctionDef* function_def = doperation.function_def; in PrepareGraphForMlir() local
942 Status MaybeInsertIdentityNodes(const FunctionDef* function_def, Graph* graph) { in MaybeInsertIdentityNodes()
976 void AddDTensorFunctionAttr(FunctionDef& function_def) { in AddDTensorFunctionAttr()
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/
H A Dshape_refiner.cc157 Status ShapeRefiner::InferShapesForFunction(const FunctionDef* function_def, in InferShapesForFunction()
697 const FunctionDef* function_def = in RunShapeFn() local
H A Doptimization_registry.h76 const FunctionDef* function_def = nullptr; member
/aosp_15_r20/external/tensorflow/tensorflow/c/experimental/saved_model/core/revived_types/
H A Dtf_concrete_function.cc41 const FunctionDef* function_def, in Create()
H A Dtf_signature_def_function.cc42 const FunctionDef* function_def, in Create()
H A Dflat_tensor_function.cc49 const FunctionDef* function_def, in Create()
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/eager/
H A Deager_operation_test.cc58 tensorflow::FunctionDef function_def; in TEST() local
H A Dkernel_and_device.cc143 const FunctionDef* function_def; in InstantiateFunc() local
H A Dexecute.cc271 const FunctionDef* function_def = in GetOutputDTypes() local
397 const FunctionDef* function_def = in GetFuncAttr() local
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/
H A Dfunction_api_info.cc29 Status FunctionApiInfo::Init(const FunctionDef& function_def) { in Init()
/aosp_15_r20/external/tensorflow/tensorflow/core/framework/
H A Dgraph_def_util_test.cc291 FunctionDef* function_def = graph_def.mutable_library()->add_function(); in TEST() local
/aosp_15_r20/external/tensorflow/tensorflow/core/protobuf/
H A Deager_service.proto211 FunctionDef function_def = 1; field

12