Searched refs:b_t_transposed_shape (Results 1 – 1 of 1) sorted by relevance
639 TensorShape b_t_transposed_shape; in Compute() local641 b_t_transposed_shape.AddDim(batch_size); in Compute()643 b_t_transposed_shape.AddDim(b_t.dim_size(row_dim + 1)); in Compute()644 b_t_transposed_shape.AddDim(b_t.dim_size(row_dim)); in Compute()646 b_t_transposed_shape, &b_t_input)); in Compute()