Home
last modified time | relevance | path

Searched defs:function_def (Results 1 – 25 of 32) 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 IsTPUGraphDef() 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/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.cc1125 string function_def = MatMulFunction(); in FunctionDefAndExecute() local
1175 string function_def = AddFunction(); in RunAddFunction() local
1277 string function_def = MatMulFunction(); in BM_ExecuteFunction() local
Dc_api.cc86 tensorflow::FunctionDef& function_def) { in AnnotateEagerRuntimeConstructionContext()
910 tensorflow::FunctionDef function_def; in TFE_ContextAddFunctionDef() local
/external/tensorflow/tensorflow/core/data/
Drewrite_utils.cc78 void AddFakeSinks(FunctionDef* function_def) { in AddFakeSinks()
93 void RemoveFakeSinks(FunctionDef* function_def) { in RemoveFakeSinks()
Ddataset_utils.cc305 const FunctionDef& function_def) { in IsFunctionStateful()
/external/tensorflow/tensorflow/core/common_runtime/
Doptimization_registry.h76 const FunctionDef* function_def = nullptr; member
Dshape_refiner.cc158 const FunctionDef* function_def, AttrSlice attributes, in InferShapesForFunction()
693 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/core/grappler/optimizers/data/
Dfunction_utils.cc184 const FunctionDef& function_def, bool skip_assert) { in IsFunctionStateful()
/external/tensorflow/tensorflow/core/protobuf/
Deager_service.proto211 FunctionDef function_def = 1; field
/external/tensorflow/tensorflow/core/framework/
Dgraph_def_util_test.cc291 FunctionDef* function_def = graph_def.mutable_library()->add_function(); in TEST() local
Dfunction.cc999 const Tensor& input, const int arg_index, const FunctionDef& function_def, in GetFunctionResourceInputDevice()
1307 std::shared_ptr<FunctionDefAndOpRegistration> function_def; in CopyFunctionDefFrom() local
/external/tensorflow/tensorflow/core/common_runtime/eager/
Dkernel_and_device.cc143 const FunctionDef* function_def; in InstantiateFunc() local
Dexecute.cc259 const FunctionDef* function_def = in GetOutputDTypes() local
349 const FunctionDef* function_def = in GetFuncAttr() local
/external/tensorflow/tensorflow/lite/python/
Dop_hint.py800 def _find_children_hints_in_while_loop(function_def, nodes_mapping): argument
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/
Dexport_graphdef.cc788 FuncOp func, const GraphExportConfig& configs, FunctionDef* function_def) { in ConvertMlirFunctionToFunctionLibraryDef()

12