Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 6 of 6 for TensorInfo (0.28 sec)

  1. 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)
  2. 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)
  3. 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)
  4. 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)
  5. 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)
  6. 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)
Back to top