• Home
  • Raw
  • Download

Lines Matching refs:fileStream

204                                      std::ofstream& fileStream);
209 …d DumpTensorElement(const TensorType& tensor, unsigned int elementIndex, std::ofstream& fileStream) in DumpTensorElement() argument
212 fileStream << static_cast<PrintableType>(elements[elementIndex]) << " "; in DumpTensorElement()
226 std::ofstream fileStream; in DumpTensor() local
227 fileStream.open(fileName.c_str(), std::ofstream::out | std::ofstream::trunc); in DumpTensor()
229 if (!fileStream.good()) in DumpTensor()
282fileStream << "Cannot dump tensor elements: not all dimensions are specified" << std::endl; in DumpTensor()
285 fileStream << "# Number of elements " << tensor.GetNumElements() << std::endl; in DumpTensor()
289 fileStream << "# Shape []" << std::endl; in DumpTensor()
292 fileStream << "# Shape [" << shape[0]; in DumpTensor()
295 fileStream << "," << shape[d]; in DumpTensor()
297 fileStream << "]" << std::endl; in DumpTensor()
298fileStream << "Each line contains the data of each of the elements of dimension0. In NCHW and NHWC… in DumpTensor()
319 (*dumpElementFunction)(tensor, elementIndex, fileStream); in DumpTensor()
323 fileStream << std::endl; in DumpTensor()
326 fileStream << std::endl; in DumpTensor()
330 fileStream << "Cannot dump tensor elements: Unsupported data type " in DumpTensor()
334 if (!fileStream.good()) in DumpTensor()
374 std::ofstream fileStream; in DumpJsonProfilingIfRequired() local
375 fileStream.open(fileName.c_str(), std::ofstream::out | std::ofstream::trunc); in DumpJsonProfilingIfRequired()
377 if (!fileStream.good()) in DumpJsonProfilingIfRequired()
384 profiler->Print(fileStream); in DumpJsonProfilingIfRequired()
411 std::ofstream fileStream; in ExportNetworkGraphToDotFile() local
412 fileStream.open(fileName, std::ofstream::out | std::ofstream::trunc); in ExportNetworkGraphToDotFile()
414 if (!fileStream.good()) in ExportNetworkGraphToDotFile()
420 if (optimizedNetwork.SerializeToDot(fileStream) != armnn::Status::Success) in ExportNetworkGraphToDotFile()