Searched defs:temp_shape (Results 1 – 3 of 3) sorted by relevance
178 TensorShape temp_shape({k, m}); in ComputeMatrix() local217 TensorShape temp_shape({static_cast<int64_t>(buffer_size)}); in SolveWithGtsv() local320 TensorShape temp_shape({static_cast<int64_t>(buffer_size)}); in ComputeWithGtsvBatched() local
224 TensorShape temp_shape{input_dims[0]}; in DoRealForwardFFT() local
3562 SDF_Shape temp_shape; /* temporary shape */ in sdf_generate_with_overlaps() local