Searched refs:temp_shape (Results 1 – 3 of 3) sorted by relevance
172 TensorShape temp_shape({k, m}); in ComputeMatrix() local175 DataTypeToEnum<Scalar>::value, temp_shape, &temp)); in ComputeMatrix()211 TensorShape temp_shape({static_cast<int64>(buffer_size)}); in SolveWithGtsv() local213 context->allocate_temp(DT_UINT8, temp_shape, &temp_tensor)); in SolveWithGtsv()319 TensorShape temp_shape({static_cast<int64>(buffer_size)}); in ComputeWithGtsvBatched() local321 context->allocate_temp(DT_UINT8, temp_shape, &temp_tensor)); in ComputeWithGtsvBatched()
219 TensorShape temp_shape{input_dims[0]}; in DoRealForwardFFT() local222 temp_shape.AddDim(fft_shape[i - 1]); in DoRealForwardFFT()231 temp_shape, &temp)); in DoRealForwardFFT()
372 temp_shape = (375 if temp_shape[-1] % 4 != 0:379 temp_shape[-1] //= 4380 temp_shape.append(4)381 permutation = permutations[len(temp_shape)]383 t = array_ops.reshape(input_shape_or_tensor, temp_shape)386 return [temp_shape[a] for a in permutation]