Searched defs:temp_shape (Results 1 – 2 of 2) sorted by relevance
172 TensorShape temp_shape({k, m}); in ComputeMatrix() local211 TensorShape temp_shape({static_cast<int64>(buffer_size)}); in SolveWithGtsv() local319 TensorShape temp_shape({static_cast<int64>(buffer_size)}); in ComputeWithGtsvBatched() local
219 TensorShape temp_shape{input_dims[0]}; in DoRealForwardFFT() local