Searched defs:reshaped_input (Results 1 – 8 of 8) sorted by relevance
417 Tensor reshaped_input = at::from_blob( in dequantize_per_token_out() local435 Tensor reshaped_input(&reshaped_input_impl); in dequantize_per_token_out() local
399 Tensor reshaped_input = at::from_blob( in quantize_per_token_out() local418 Tensor reshaped_input(&reshaped_input_impl); in quantize_per_token_out() local
161 HloInstruction* reshaped_input = computation->AddInstruction( in ExpandInstruction() local
219 const Tensor reshaped_input = self.reshape({M, N}); in histogramdd_out_cpu_template() local
493 auto reshaped_input = in matchAndRewrite() local
412 const auto reshaped_input = reshape_dim_outof(1, group, input); in native_group_norm_backward_plumbing() local
1175 Value reshaped_input = builder.create<ReshapeOp>( in BuildReshapeOp() local
1792 Value reshaped_input = rewriter.create<mhlo::ReshapeOp>( in matchAndRewrite() local