transpose_folding_test.cc | 64 HloInstruction* transpose_y = local 72 /*rhs=*/transpose_y, dot_dnums)); 175 HloInstruction* transpose_y = local 183 /*rhs=*/transpose_y, dot_dnums)); 190 {transpose_y, dot}, "outlined", entry_computation); 222 HloInstruction* transpose_y = local 235 transpose_y->shape().dimensions(dnums.kernel_spatial_dimensions(i))); 238 x->shape(), transpose_y->shape(), window, dnums); 241 conv_shape.ValueOrDie(), x, transpose_y, window, dnums)); 275 HloInstruction* transpose_y local [all...] |