Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 3 of 3 for ConvertTfliteConstTensor (0.29 sec)

  1. tensorflow/compiler/mlir/lite/utils/const_tensor_utils.h

        bool truncate = false);
    
    absl::StatusOr<mlir::ElementsAttr> ConvertFloatBuffer(
        mlir::RankedTensorType shaped_type, const std::vector<uint8_t>& buffer);
    
    tensorflow::TensorProto ConvertTfliteConstTensor(
        const tflite::TensorT& tensor, const std::vector<uint8_t>& buffer);
    
    }  // namespace TFL
    }  // namespace mlir
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 07 23:04:40 UTC 2024
    - 2.9K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/lite/utils/const_tensor_utils.cc

        }
      }
      return absl::InvalidArgumentError(absl::StrCat(
          "unsupported bit width ", elem_type.getIntOrFloatBitWidth()));
    }
    
    tensorflow::TensorProto ConvertTfliteConstTensor(
        const TensorT& tensor, const std::vector<uint8_t>& buffer) {
      tensorflow::TensorProto ret;
      ret.set_dtype(TflTypeToTfType(tensor.type));
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 07 23:04:40 UTC 2024
    - 16.6K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/lite/flatbuffer_import.cc

                                                   /*get_storage=*/true));
      auto value_type = mlir::dyn_cast<mlir::RankedTensorType>(type);
    
      tensorflow::TensorProto repr = tfl::ConvertTfliteConstTensor(tensor, buffer);
      repr.clear_tensor_shape();
      if (tfl::IsQuantized(tensor)) {
        repr.mutable_tensor_shape()->add_dim()->set_size(buffer.size());
        repr.set_dtype(tensorflow::DT_INT8);
      } else {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 21 18:21:50 UTC 2024
    - 66.8K bytes
    - Viewed (0)
Back to top