Searched refs:getTotalTensorBufferSize (Results 1 – 5 of 5) sorted by relevance
163 InputSpecs[I].getTotalTensorBufferSize()); in TFModelEvaluatorImpl() 202 if (Spec.getTotalTensorBufferSize() != Tensor->bytes) in checkReportAndInvalidate()
32 OutputBuffer(OutputSpec.getTotalTensorBufferSize()) { in InteractiveModelRunner()
101 OS->write(RawData, Spec.getTotalTensorBufferSize()); in writeTensor()
66 OwnedBuffers.emplace_back(Spec.getTotalTensorBufferSize()); in setUpBufferForTensor()
88 size_t getTotalTensorBufferSize() const { return ElementCount * ElementSize; } in getTotalTensorBufferSize() function