Searched defs:x_col (Results 1 – 14 of 14) sorted by relevance
57 int64_t x_col = x_shp[kIndex1]; in InferShape() local105 int64_t x_col = x_shape[kIndex1]; in InferShape() local
72 int64_t x_col = x_shp[2]; in InferShape() local139 int64_t x_col = x_shp[2]; in InferShape() local
34 auto x_col = x_shape_vec[x_rank - kIndex2]; in InferShape() local
89 auto x_col = x_shp[x_shp.size() - 1]; in InferShape() local193 auto x_col = x_shp[x_shp.size() - 1]; in InferShape() local
38 auto x_col = x_shape_vec[x_rank - kIndex2]; in InferShape() local
57 int64_t x_col = x_shp[1]; in InferShape() local
57 int64_t x_col = x_shp[0]; in InferShape() local
40 auto x_col = x_shape_vec[x_rank - kIndex2]; in InferShape() local
74 auto x_col = x_shp[(transpose_a ? 0 : 1)]; in InferShape2D() local194 auto x_col = x_shp[(transpose_a ? 0 : 1)]; in InferShape() local
74 int64_t x_col = x_mat_shape[static_cast<size_t>(!transpose_a)]; in CheckBatchMatmulInputWhetherCanBeMul() local
161 int64_t x_col = x_mat_shape[static_cast<size_t>(!transpose_x)]; in CheckBatchMatmulInputWhetherCanBeMul() local
76 auto x_col = x_shp[(transpose_a ? 0 : 1)]; in MatMulAllReduceInferShape() local
331 auto x_col = x_shape[3]; in Calc() local
990 static int x_col; /* current column on line */ variable