Searched refs:activation_input_feature_dim (Results 1 – 1 of 1) sorted by relevance
387 int64 activation_input_feature_dim = dim_numbers.input_feature_dimension(); in HandleConvolution() local411 reshaped_activation_shape.set_dimensions(activation_input_feature_dim, in HandleConvolution()493 int64 activation_input_feature_dim = in HandleConvolution() local507 activation_slice_starts[activation_input_feature_dim] = i * group_size; in HandleConvolution()508 activation_slice_limits[activation_input_feature_dim] = in HandleConvolution()511 activation_sliced_shape.set_dimensions(activation_input_feature_dim, in HandleConvolution()