Searched refs:output_batch_dim (Results 1 – 3 of 3) sorted by relevance
101 auto output_batch_dim = dnums.output_batch_dimension(); in Run() local103 new_output_dim_order[0] = output_batch_dim; in Run()104 new_conv_dims[0] = hlo->shape().dimensions(output_batch_dim); in Run()
87 auto output_batch_dim = conv_dnums.output_batch_dimension(); in MatchBackwardFilter() local114 if (input_batch_dim == output_batch_dim && in MatchBackwardFilter()205 backward_conv_dnums.set_kernel_input_feature_dimension(output_batch_dim); in MatchBackwardFilter()
933 const int64 output_batch_dim = dnums.output_batch_dimension(); in HandleConvolution() local957 lhs_index[input_batch_dim] = out_index[output_batch_dim]; in HandleConvolution()