Lines Matching refs:output_flag

126       int output_flag = 0;  in NumpyUniqueCPUNoneAxisImpl()
128 output_flag += 1; in NumpyUniqueCPUNoneAxisImpl()
129 const_cast<NDArray &>(outputs[output_flag]).Init(s); in NumpyUniqueCPUNoneAxisImpl()
130 dim_t* unique_indices = outputs[output_flag].data().dptr<dim_t>(); in NumpyUniqueCPUNoneAxisImpl()
137 output_flag += 1; in NumpyUniqueCPUNoneAxisImpl()
138 const_cast<NDArray &>(outputs[output_flag]).Init(mxnet::TShape(1, input_size)); in NumpyUniqueCPUNoneAxisImpl()
139 dim_t* unique_inverse = outputs[output_flag].data().dptr<dim_t>(); in NumpyUniqueCPUNoneAxisImpl()
144 output_flag += 1; in NumpyUniqueCPUNoneAxisImpl()
154 const_cast<NDArray &>(outputs[output_flag]).Init(s); in NumpyUniqueCPUNoneAxisImpl()
155 dim_t* unique_counts = outputs[output_flag].data().dptr<dim_t>(); in NumpyUniqueCPUNoneAxisImpl()
242 int output_flag = 0; in NumpyUniqueCPUImpl()
244 output_flag += 1; in NumpyUniqueCPUImpl()
245 const_cast<NDArray &>(outputs[output_flag]).Init(mxnet::TShape(1, valid_num)); in NumpyUniqueCPUImpl()
246 dim_t* unique_indices = outputs[output_flag].data().dptr<dim_t>(); in NumpyUniqueCPUImpl()
253 output_flag += 1; in NumpyUniqueCPUImpl()
254 const_cast<NDArray &>(outputs[output_flag]).Init(mxnet::TShape(1, temp_shape[0])); in NumpyUniqueCPUImpl()
255 dim_t* unique_inverse = outputs[output_flag].data().dptr<dim_t>(); in NumpyUniqueCPUImpl()
260 output_flag += 1; in NumpyUniqueCPUImpl()
270 const_cast<NDArray &>(outputs[output_flag]).Init(mxnet::TShape(1, valid_num)); in NumpyUniqueCPUImpl()
271 dim_t* unique_counts = outputs[output_flag].data().dptr<dim_t>(); in NumpyUniqueCPUImpl()
298 int output_flag = 0; in NumpyUniqueCPUForward() local
300 output_flag += 1; in NumpyUniqueCPUForward()
301 const_cast<NDArray &>(outputs[output_flag]).Init(shape_1); in NumpyUniqueCPUForward()
303 outputs[output_flag].data().get_with_shape<cpu, 1, dim_t>(Shape1(1), s); in NumpyUniqueCPUForward()
307 output_flag += 1; in NumpyUniqueCPUForward()
308 const_cast<NDArray &>(outputs[output_flag]).Init(shape_1); in NumpyUniqueCPUForward()
310 outputs[output_flag].data().get_with_shape<cpu, 1, dim_t>(Shape1(1), s); in NumpyUniqueCPUForward()
314 output_flag += 1; in NumpyUniqueCPUForward()
315 const_cast<NDArray &>(outputs[output_flag]).Init(shape_1); in NumpyUniqueCPUForward()
317 outputs[output_flag].data().get_with_shape<cpu, 1, dim_t>(Shape1(1), s); in NumpyUniqueCPUForward()
331 int output_flag = 0; in NumpyUniqueCPUForward() local
333 output_flag += 1; in NumpyUniqueCPUForward()
334 const_cast<NDArray &>(outputs[output_flag]).Init(shape_0); in NumpyUniqueCPUForward()
337 output_flag += 1; in NumpyUniqueCPUForward()
338 const_cast<NDArray &>(outputs[output_flag]).Init(shape_0); in NumpyUniqueCPUForward()
341 output_flag += 1; in NumpyUniqueCPUForward()
342 const_cast<NDArray &>(outputs[output_flag]).Init(shape_0); in NumpyUniqueCPUForward()