Searched refs:row_dim (Results 1 – 2 of 2) sorted by relevance
328 int64 row_dim = dim_nums.lhs_batch_dimensions_size(); in ExecuteOnStream() local336 CHECK_NE(row_dim, batch_dim); in ExecuteOnStream()343 CHECK_LT(shape->layout().minor_to_major(row_dim), 2); in ExecuteOnStream()351 int64 output_num_rows = output_shape_.dimensions(row_dim); in ExecuteOnStream()377 bool is_row_major = LayoutUtil::Minor(shape.layout(), row_dim) != 0; in ExecuteOnStream()378 bool layout_mismatch = LayoutUtil::Minor(shape.layout(), row_dim) != in ExecuteOnStream()379 LayoutUtil::Minor(output_shape_.layout(), row_dim); in ExecuteOnStream()382 shape.dimensions(row_dim + static_cast<int64>(is_row_major)), in ExecuteOnStream()383 shape.dimensions(row_dim + static_cast<int64>(!is_row_major)), in ExecuteOnStream()388 lhs_data, lhs_shape_, dim_nums.lhs_contracting_dimensions(0) == row_dim); in ExecuteOnStream()[all …]
249 for (int64_t row_dim = strides.size() - 2; row_dim >= 0; in GetNextInternal() local250 --row_dim) { in GetNextInternal()251 strides[row_dim] = in GetNextInternal()252 strides[row_dim + 1] * t.shape().dim_size(row_dim + 1); in GetNextInternal()