Searched refs:s_shape (Results 1 – 5 of 5) sorted by relevance
123 s_shape = array_ops.shape(self.spectrum)124 self._block_shape_tensor = s_shape[-self.block_depth:]309 s_shape = self._spectrum.get_shape()317 batch_shape = s_shape[:-self.block_depth]319 trailing_dims = s_shape[-self.block_depth:]329 s_shape = array_ops.shape(self._spectrum)330 batch_shape = s_shape[:-self.block_depth]331 trailing_dims = s_shape[-self.block_depth:]
359 s_shape = tensors[0].shape361 s_shape = array_ops.broadcast_static_shape(s_shape, t.shape)362 if s_shape.is_fully_defined():363 return s_shape.as_list()
419 s_shape = array_ops.shape(s)420 eye = _linalg.eye(s_shape[-1], batch_shape=s_shape[:-1], dtype=s.dtype)
191 TensorShape s_shape = in Compute() local193 size_t s_dims = s_shape.dims(); in Compute()206 size_t s_size = s_shape.dim_size(d); in Compute()
1607 Shape s_shape = ShapeUtil::MakeShape(indices_type, {1}); in OptimizeDotOfGather() local