Searched defs:output_fp32 (Results 1 – 1 of 1) sorted by relevance
263 Tensor output_fp32 = at::zeros({output_size, ddim}, output.options().dtype(at::kFloat)); in index_select_add() local327 Tensor output_fp32 = in index_select_add() local642 Tensor output_fp32 = in index_select_scale_add() local708 Tensor output_fp32 = in index_select_scale_add() local