- Sort Score
- Result 10 results
- Languages All
Results 1 - 6 of 6 for TensorInfo (0.28 sec)
-
tensorflow/compiler/mlir/quantization/tensorflow/python/representative_dataset_test.py
@test_util.deprecated_graph_mode_only def test_create_feed_dict_from_input_data(self): signature_def = meta_graph_pb2.SignatureDef( inputs={'input_tensor': meta_graph_pb2.TensorInfo(name='input:0')} ) rng = np.random.default_rng(seed=14) input_tensor_value = rng.random(size=(2, 2)) sample = {'input_tensor': input_tensor_value}
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jan 04 07:35:19 UTC 2024 - 11.6K bytes - Viewed (0) -
tensorflow/cc/tools/freeze_saved_model_test.cc
SavedModelBundle saved_model_bundle; SignatureDef signature_def; TensorInfo& in = (*signature_def.mutable_inputs())["input_arg"]; in.mutable_composite_tensor()->add_components()->set_name("input1:0"); in.mutable_composite_tensor()->add_components()->set_name("input2:0"); TensorInfo& out = (*signature_def.mutable_outputs())["output_arg"];
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 07 13:30:31 UTC 2022 - 21.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/tf_saved_model/no_input_shape_v1.py
def Test(): x = tf.placeholder(dtype=tf.float32, shape=[None]) batch_size = tf.shape(x)[0] r = tf.convert_to_tensor([batch_size, 1]) tensor_info_x = meta_graph_pb2.TensorInfo( name=x.name, dtype=tf.as_dtype(x.dtype).as_datatype_enum) tensor_info_r = tf.compat.v1.saved_model.utils.build_tensor_info(r) return {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Sep 28 21:37:05 UTC 2021 - 2.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/cc/convert_asset_args.cc
const StringRef tensor_name) { AssetFileDef asset_file_def{}; asset_file_def.set_filename(MaybeStripAssetDirectoryPrefix(filename).str()); tensorflow::TensorInfo tensor_info{}; tensor_info.set_name(tensor_name.str()); *asset_file_def.mutable_tensor_info() = tensor_info; return asset_file_def; } // Returns a list of "tf.entry_function" attribute's "inputs" comma-split
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 5.4K bytes - Viewed (0) -
tensorflow/cc/tools/freeze_saved_model.cc
#include "tensorflow/core/protobuf/meta_graph.pb.h" namespace tensorflow { namespace { // Gets tensor names from tensor_info and inserts them into the set of tensor // names. void GetTensorNamesFromTensorInfo(const TensorInfo& tensor_info, std::unordered_set<string>* tensor_names) { if (tensor_info.has_coo_sparse()) { // If the tensor is sparse we have to add all three tensors of the sparse
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 11 08:05:36 UTC 2023 - 11.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/translate/import_model.cc
context.getOrLoadDialect(name); } absl::StatusOr<std::string> GetDenseTensorNameFromTensorInfo( const TensorInfo& tensor_info) { // TODO(b/184675681): Support other encoding cases. // // TODO(b/184679394): Add unit test for this check. TF_RET_CHECK(tensor_info.encoding_case() == tensorflow::TensorInfo::kName) << "Only dense tensor is supported, but got encoding case " << tensor_info.encoding_case();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 01 11:17:36 UTC 2024 - 183.2K bytes - Viewed (0)