Home
last modified time | relevance | path

Searched refs:unwrapTensorAtLevel (Results 1 – 11 of 11) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/functorch/
H A DBatchRulesNorm.cpp228 auto [grad_out_value, grad_out_bdim] = unwrapTensorAtLevel(grad_out, cur_level); in batch_norm_backward_plumbing()
229 auto [input_value, input_bdim] = unwrapTensorAtLevel(input, cur_level); in batch_norm_backward_plumbing()
234 std::tie(weight_value, weight_bdim) = unwrapTensorAtLevel(weight, cur_level); in batch_norm_backward_plumbing()
239 std::tie(running_mean_value, running_mean_bdim) = unwrapTensorAtLevel(running_mean, cur_level); in batch_norm_backward_plumbing()
244 std::tie(running_var_value, running_var_bdim) = unwrapTensorAtLevel(running_var, cur_level); in batch_norm_backward_plumbing()
246 auto [save_mean_value, save_mean_bdim] = unwrapTensorAtLevel(save_mean, cur_level); in batch_norm_backward_plumbing()
247 auto [save_rstd_value, save_rstd_bdim] = unwrapTensorAtLevel(save_rstd, cur_level); in batch_norm_backward_plumbing()
270 … unwrapTensorAtLevel(grad_normalized_input.transpose(0, 1), cur_level); // [B0, B, C, *] in batch_norm_backward_plumbing()
305 auto [input_value, input_bdim] = unwrapTensorAtLevel(input, cur_level); in native_group_norm_plumbing()
392 auto [input_value, input_bdim] = unwrapTensorAtLevel(input, cur_level); in native_group_norm_backward_plumbing()
[all …]
H A DBatchRulesLoss.cpp101 auto [self_value, self_bdim] = unwrapTensorAtLevel(self, cur_level); in binary_cross_entropy_plumbing()
102 auto [target_value, target_bdim] = unwrapTensorAtLevel(target, cur_level); in binary_cross_entropy_plumbing()
136 auto [grad_value, grad_bdim] = unwrapTensorAtLevel( in binary_cross_entropy_backward_plumbing()
138 auto [input_value, input_bdim] = unwrapTensorAtLevel(input, cur_level); in binary_cross_entropy_backward_plumbing()
139 auto [target_value, target_bdim] = unwrapTensorAtLevel(target, cur_level); in binary_cross_entropy_backward_plumbing()
H A DBatchRulesRandomness.cpp42 auto [self_value, self_bdim] = unwrapTensorAtLevel(self, cur_level); in random_inplace_batching_rule()
67 auto [self_value, self_bdim] = unwrapTensorAtLevel(self, cur_level); in bernoulli_inplace_Tensor_batching_rule()
69 auto [other_value, other_bdim] = unwrapTensorAtLevel(p_, cur_level); in bernoulli_inplace_Tensor_batching_rule()
131 auto [tensor_value, tensor_bdim] = unwrapTensorAtLevel(tensor, cur_level); in unary_pointwise_random_batch_rule()
159 auto [tensor_value, tensor_bdim] = unwrapTensorAtLevel(self, cur_level); in tensor_like_random_batch_rule()
182 auto [tensor_value, tensor_bdim] = unwrapTensorAtLevel(tensor, cur_level); in native_dropout_batching_rule()
221 auto [self_value, self_bdim] = unwrapTensorAtLevel(self, cur_level); in multinomial_batching_rule()
H A DPlumbingHelper.cpp43 std::tuple<Tensor, std::optional<int64_t>> unwrapTensorAtLevel(const Tensor& tensor, int64_t level)… in unwrapTensorAtLevel() function
55 auto result = unwrapTensorAtLevel(tensor, level); in isBatchedAtLevel()
H A DBatchRulesConvolution.cpp377 auto [grad_output, grad_output_bdim] = unwrapTensorAtLevel(grad_output_, cur_level); in convolution_backward_plumbing()
378 auto [input, input_bdim] = unwrapTensorAtLevel(input_, cur_level); in convolution_backward_plumbing()
379 auto [weight, weight_bdim] = unwrapTensorAtLevel(weight_, cur_level); in convolution_backward_plumbing()
H A DBatchRulesScatterOps.cpp328 auto [self_value, self_bdim] = unwrapTensorAtLevel(self, cur_level); in index_plumbing()
336 std::tie(index, index_bdim) = unwrapTensorAtLevel(ind.value(), cur_level); in index_plumbing()
458 auto [self_value, self_bdim] = unwrapTensorAtLevel(self, cur_level); in unpackSelfAndIndicesAndValuesAtCurrentLevel()
467 std::tie(index, index_bdim) = unwrapTensorAtLevel(ind.value(), cur_level); in unpackSelfAndIndicesAndValuesAtCurrentLevel()
472 auto [values_value, values_bdim] = unwrapTensorAtLevel(values, cur_level); in unpackSelfAndIndicesAndValuesAtCurrentLevel()
H A DPlumbingHelper.h38 TORCH_API std::tuple<Tensor, std::optional<int64_t>> unwrapTensorAtLevel(const Tensor& tensor, int6…
H A DBatchRulesHelper.h147 auto [tensor_value, tensor_bdim] = unwrapTensorAtLevel(ivalue.toTensor(), cur_level);
222 auto unpacked = unwrapTensorAtLevel(ivalue.toTensor(), cur_level);
H A DBatchRulesBinaryOps.cpp63 auto [tensor_value, tensor_bdim] = unwrapTensorAtLevel(tensor, cur_level); in apply()
65 auto [other_value, other_bdim] = unwrapTensorAtLevel(other, cur_level); in apply()
H A DBatchRulesReduceOps.cpp126 auto [self, self_bdim] = unwrapTensorAtLevel(arguments[0].toTensor(), cur_level); in boxed_reduction_batch_rule()
H A DBatchRulesViews.cpp165 auto [self_value, self_bdim] = unwrapTensorAtLevel(self, cur_level); in resize__plumbing()