Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 1 of 1 for ConvertFlatTensor (0.24 sec)

  1. tensorflow/compiler/mlir/tensorflow/utils/convert_tensor.cc

      return mangling_util::MangleTensor(ConvertToProto(tensor));
    }
    
    // Converts a TensorFlow tensor into an MLIR elements attribute.
    template <typename T>
    absl::StatusOr<ElementsAttr> ConvertFlatTensor(const Tensor& input_tensor,
                                                   ShapedType type) {
      auto arr = input_tensor.flat<T>();
      return ElementsAttr(mlir::DenseElementsAttr::get(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Apr 26 09:37:10 UTC 2024
    - 20.5K bytes
    - Viewed (0)
Back to top