Searched defs:row_dim (Results 1 – 7 of 7) sorted by relevance
298 const int row_dim = a.dense_shape_host.size() == 2 ? 0 : 1; in GetWorkspaceSize() local326 const int row_dim = a.dense_shape_host.size() == 2 ? 0 : 1; in GetOutputStructure() local354 const int row_dim = a.dense_shape_host.size() == 2 ? 0 : 1; in Compute() local
245 const int row_dim = (rank == 2) ? 0 : 1; in ValidateInputs() local
138 const int row_dim = (rank == 2) ? 0 : 1; in Compute() local343 const int row_dim = (rank == 2) ? 0 : 1; in Compute() local
518 const int row_dim = (rank == 2) ? 0 : 1; in Compute() local
221 int64_t row_dim = batch_dimensions.size(); in RunGemm() local263 int64_t stride) -> MatrixDescriptor { in RunGemm()
257 for (int64_t row_dim = strides.size() - 2; row_dim >= 0; in GetNextInternal() local
226 int64_t row_dim = dim_nums.lhs_batching_dimensions().size(); in GemmOpConversionRewrite() local