Searched refs:tmp_in (Results 1 – 2 of 2) sorted by relevance
79 std::vector<T> tmp_in( out_values, out_values + n); in all_reduce_impl() local80 reduce(comm, detail::c_data(tmp_in), n, out_values, op, 0); in all_reduce_impl()
243 …auto tmp_in = src_in + batch_index * conv_param_->input_h_ * conv_param_->input_w_ * conv_param_->… in Run() local245 Conv1x1InputPack(tmp_in, input_ptr_, conv_param_, sizeof(float)); in Run()247 input_ptr_ = tmp_in; in Run()