Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/core/grappler/utils/
Dtpu_test.cc34 FunctionDef* function_def = library->add_function(); in TEST_F() local
55 FunctionDef* function_def = library->add_function(); in TEST_F() local
Dtpu.cc31 for (const auto& function_def : def.library().function()) { in IsLegacyTPUBridgeGraphDef() local
/external/tensorflow/tensorflow/c/experimental/saved_model/core/
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
/external/tensorflow/tensorflow/core/grappler/optimizers/inference/
Dbatch_op_rewriter.cc165 FunctionDef* function_def = in Optimize() local
196 FunctionDef* function_def = in Optimize() local
Dbatch_op_rewriter_test.cc77 FunctionDef* function_def = function_def_lib->add_function(); local
/external/tensorflow/tensorflow/c/experimental/saved_model/core/revived_types/
Dtf_concrete_function.cc41 const FunctionDef* function_def, in Create()
Dtf_signature_def_function.cc42 const FunctionDef* function_def, in Create()
Dflat_tensor_function.cc49 const FunctionDef* function_def, in Create()
/external/tensorflow/tensorflow/core/common_runtime/eager/
Deager_operation_test.cc58 tensorflow::FunctionDef function_def; in TEST() local
Dkernel_and_device.cc143 const FunctionDef* function_def; in InstantiateFunc() local
/external/tensorflow/tensorflow/c/eager/
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
Dc_api_remote_test_util.cc140 string function_def = MatMulFunction(matmul_device); in TestRemoteExecuteSilentCopies() local
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
/external/tensorflow/tensorflow/core/grappler/optimizers/data/
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
Dfunction_utils.cc184 const FunctionDef& function_def, bool skip_assert) { in IsFunctionStateful()
/external/tensorflow/tensorflow/core/data/
Drewrite_utils.cc80 void AddFakeSinks(FunctionDef* function_def) { in AddFakeSinks()
95 void RemoveFakeSinks(FunctionDef* function_def) { in RemoveFakeSinks()
/external/tensorflow/tensorflow/core/common_runtime/
Doptimization_registry.h76 const FunctionDef* function_def = nullptr; member
Dshape_refiner.cc157 Status ShapeRefiner::InferShapesForFunction(const FunctionDef* function_def, in InferShapesForFunction()
697 const FunctionDef* function_def = in RunShapeFn() local
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dfunction_api_info.cc29 Status FunctionApiInfo::Init(const FunctionDef& function_def) { in Init()
/external/tensorflow/tensorflow/dtensor/cc/
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()
Ddtensor_device_util.h129 const FunctionDef* function_def; member
/external/tensorflow/tensorflow/core/framework/
Dgraph_def_util_test.cc291 FunctionDef* function_def = graph_def.mutable_library()->add_function(); in TEST() local
/external/tensorflow/tensorflow/core/protobuf/
Deager_service.proto211 FunctionDef function_def = 1; field
/external/tensorflow/tensorflow/core/tfrt/utils/
Dtfrt_graph_execution_state.cc85 for (const auto& function_def : graph_def.library().function()) { in FindFunctionsToOptimize() local
/external/tensorflow/tensorflow/lite/python/
Dop_hint.py795 def _find_children_hints_in_while_loop(function_def, nodes_mapping): argument

12