- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for mutable_signature (0.56 sec)
-
tensorflow/c/experimental/saved_model/core/tf_concrete_function_loading_test.cc
testing::CreateTensorHandle(ctx, DT_FLOAT, {2, 4}, value)) {} }; FunctionDef FuncDefWithNumInputsOutputs(int num_inputs, int num_outputs) { FunctionDef func; OpDef* signature = func.mutable_signature(); for (int i = 0; i < num_inputs; ++i) { signature->add_input_arg(); } for (int i = 0; i < num_outputs; ++i) { signature->add_output_arg(); } return func; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Apr 14 19:16:58 UTC 2023 - 10.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/translate/export_graphdef.cc
} auto stateful_string = mlir::TF::TensorFlowDialect::GetStatefulAttrName(); if (auto attr = function->getAttrOfType<mlir::UnitAttr>(stateful_string)) { func_def.mutable_signature()->set_is_stateful(true); } // Ignore the gradient and is_stateful attribute on the function as they have // been handled above. Ignore the entry func attribute as it is an MLIR
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 01 11:17:36 UTC 2024 - 35.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/api/v2/tf_executor_to_graph.cc
} auto stateful_string = mlir::TF::TensorFlowDialect::GetStatefulAttrName(); if (auto attr = function->getAttrOfType<mlir::UnitAttr>(stateful_string)) { func_def.mutable_signature()->set_is_stateful(true); } // Ignore the gradient and is_stateful attribute on the function as they have // been handled above. Ignore the entry func attribute as it is an MLIR
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 06 23:04:51 UTC 2024 - 35.2K bytes - Viewed (0)