Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for TensorInfo (0.21 sec)

  1. 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)
  2. 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