Searched refs:b_t_transposed_shape (Results 1 – 1 of 1) sorted by relevance
634 TensorShape b_t_transposed_shape; in Compute() local636 b_t_transposed_shape.AddDim(batch_size); in Compute()638 b_t_transposed_shape.AddDim(b_t.dim_size(row_dim + 1)); in Compute()639 b_t_transposed_shape.AddDim(b_t.dim_size(row_dim)); in Compute()641 b_t_transposed_shape, &b_t_input)); in Compute()