Searched refs:a_contracting_dimension (Results 1 – 1 of 1) sorted by relevance
1519 [](HloInstruction* a, HloInstruction* b, int a_contracting_dimension) { in OptimizeDotOfGather() argument1534 if (dynamic_slice_shape.dimensions(1 - a_contracting_dimension) != 1) { in OptimizeDotOfGather()1541 if (dynamic_slice_op_shape.dimensions(a_contracting_dimension) != in OptimizeDotOfGather()1542 dynamic_slice_shape.dimensions(a_contracting_dimension)) { in OptimizeDotOfGather()