How do I get the MatMul dimensions for GEMMs used by TensorFlow ( and PyTorch)

whatdhack February 11, 2023, 2:20am 1

Hi, TensorBoard shows MatMuls/GEMM ops, but does not show what the tensor dimensions are. Is there way to get those dimensions ?

Topic		Replies	Views
Is there any common way to check output dimensions of any layer? TensorRT	1	419	November 30, 2021
How to get the input dimensions in INetworkDefinition？ TensorRT tensorrt	4	382	June 3, 2021
Graphs mean Frameworks tensorflow	0	359	April 27, 2020
ML workload - getting matrix sizes from profiler CUDA Programming and Performance	3	38	January 23, 2025
getOutputDimensions for multiple outputs in plugin TensorRT	1	585	August 18, 2020
Jetson AGX Xavier GEMM benchmarks Jetson AGX Xavier performance	3	826	September 27, 2021
Where can I find cublasMatmulBench tool? CUDA Programming and Performance performance	2	1707	August 24, 2023
matmul with large matrices fails with float16, but succeeds with float32 Frameworks tensorflow	0	708	April 2, 2019
How can I get the tile size of gemm for cudnn kernel name? Nsight Compute	2	440	May 24, 2023
How to get the 3d dimenstion of objects in GUI Isaac Sim deployment-guides	1	26	October 30, 2024