Searched refs:inputs_rank (Results 1 – 2 of 2) sorted by relevance
1077 inputs_rank = input.shape.rank1081 elif inputs_rank:1082 num_spatial_dims = inputs_rank - 21092 if inputs_rank:1093 num_batch_dims = inputs_rank - num_spatial_dims - 1 # Channel dimension.
190 int64_t inputs_rank = -1; in Verify() local195 inputs_rank = ty.getRank(); in Verify()199 if (inputs_rank == -1) return success(); in Verify()204 int64_t range_begin = -inputs_rank - 1; // Inclusive in Verify()205 int64_t range_end = inputs_rank + 1; // Exclusive in Verify()