Searched refs:new_output_shape (Results 1 – 3 of 3) sorted by relevance
147 Shape* new_output_shape; in PadForTensorCores() local148 std::tie(new_input_shape, new_filter_shape, new_output_shape) = [&] { in PadForTensorCores()169 new_output_shape->dimensions(dnums.output_feature_dimension()); in PadForTensorCores()180 pad_dim(new_output_shape, dnums.output_feature_dimension()); in PadForTensorCores()
423 Shape new_output_shape = convolution->shape(); in HandleConvolution() local424 ShapeUtil::AppendMajorDimension(1, &new_output_shape); in HandleConvolution()443 new_output_shape, activation, filter, group_count, in HandleConvolution()
2059 new_output_shape = array_ops.concat(2061 output = array_ops.reshape(output, new_output_shape)