Searched defs:input_dim_x (Results 1 – 1 of 1) sorted by relevance
60 int64_t input_dim_x, int64_t input_dim_y, in get_index_mapping2d()127 int64_t input_dim_x, int64_t input_dim_y, in reflection_pad2d_out_kernel()147 int64_t input_dim_x, int64_t input_dim_y, in reflection_pad2d_backward_out_kernel()