Searched defs:ConvertTensor (Results 1 – 2 of 2) 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()
120 StatusOr<ElementsAttr> ConvertTensor(const Tensor& input_tensor, in ConvertTensor() function