Home
last modified time | relevance | path

Searched refs:getTotalTensorBufferSize (Results 1 – 5 of 5) sorted by relevance

/freebsd/contrib/llvm-project/llvm/lib/Analysis/
H A DTFLiteUtils.cpp163 InputSpecs[I].getTotalTensorBufferSize()); in TFModelEvaluatorImpl()
202 if (Spec.getTotalTensorBufferSize() != Tensor->bytes) in checkReportAndInvalidate()
H A DInteractiveModelRunner.cpp32 OutputBuffer(OutputSpec.getTotalTensorBufferSize()) { in InteractiveModelRunner()
/freebsd/contrib/llvm-project/llvm/include/llvm/Analysis/Utils/
H A DTrainingLogger.h101 OS->write(RawData, Spec.getTotalTensorBufferSize()); in writeTensor()
/freebsd/contrib/llvm-project/llvm/include/llvm/Analysis/
H A DMLModelRunner.h66 OwnedBuffers.emplace_back(Spec.getTotalTensorBufferSize()); in setUpBufferForTensor()
H A DTensorSpec.h88 size_t getTotalTensorBufferSize() const { return ElementCount * ElementSize; } in getTotalTensorBufferSize() function