Searched defs:ConvertTensor (Results 1 – 5 of 5) sorted by relevance
87 struct ConvertTensor { struct88 using ResultType = tensorflow::tfrt_stub::FallbackTensor;89 using ConversionContext = TensorflowConversionContext;92 static llvm::ArrayRef<int64_t> Sizes(StridedMemRefType<T, rank>* memref) { in Sizes()97 static llvm::ArrayRef<int64_t> Sizes(StridedMemRefType<T, 0>* memref) { in Sizes()102 static tensorflow::Tensor Convert(const ConversionContext& ctx, in Convert()
106 struct ConvertTensor { struct107 using ResultType = tfrt_stub::FallbackTensor;108 using ConversionContext = TensorflowConversionContext;111 static llvm::ArrayRef<int64_t> Sizes(StridedMemRefType<T, rank>* memref) { in Sizes()116 static llvm::ArrayRef<int64_t> Sizes(StridedMemRefType<T, 0>* memref) { in Sizes()121 static Tensor Convert(ConversionContext& ctx, void* memref_ptr) { in Convert()
129 StatusOr<Tensor> ConvertTensor(std::unique_ptr<tf::Tensor> tensor) { in ConvertTensor() function
116 StatusOr<ElementsAttr> ConvertTensor(const Tensor& input_tensor, in ConvertTensor() function
113 tensorflow::StatusOr<ElementsAttr> ConvertTensor(const Tensor& input_tensor, in ConvertTensor() function