Searched refs:block_shape_tensor (Results 1 – 8 of 8) sorted by relevance
96 const int64_t block_shape_tensor[NUM_BLOCK_DIMS], in operator ()()111 block_shape[block_dim] = block_shape_tensor[block_dim]; in operator ()()
287 def block_shape_tensor(self): member in _BaseLinearOperatorCirculant337 (vec_leading_shape, self.block_shape_tensor()), 0)
132 name: "block_shape_tensor"