Home
last modified time | relevance | path

Searched defs:input_reshaped (Results 1 – 20 of 20) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dsplit_v_op.cc219 const InputReshapedType& input_reshaped, in operator ()()
335 auto input_reshaped = in Compute() local
348 auto input_reshaped = input.shaped<T, 3>( in Compute() local
446 auto input_reshaped = input.shaped<T, 2>( in Compute() local
H A Dsplit_op.cc132 const InputReshapedType& input_reshaped, in operator ()()
235 auto input_reshaped = in Compute() local
248 auto input_reshaped = input.shaped<T, 3>( in Compute() local
H A Dunpack_op.cc99 auto input_reshaped = in Compute() local
H A Dlist_kernels.h1098 auto input_reshaped = Tensor(); in Compute() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/mkldnn/
H A DLinear.cpp150 auto input_reshaped = input.dim() > 2 ? input.reshape({-1, input.size(input.dim() - 1)}) : input; in mkldnn_linear_backward_weights() local
197 auto input_reshaped = in mkldnn_linear_pointwise() local
281 auto input_reshaped = in mkldnn_linear_pointwise_binary() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DPixelShuffle.cpp97 const auto input_reshaped = self.reshape(added_dims_shape); in math_pixel_shuffle() local
138 const auto input_reshaped = self.reshape(added_dims_shape); in math_pixel_unshuffle() local
H A DChanelShuffle.cpp69 auto input_reshaped = self.view({b, groups, oc, -1}); in math_channel_shuffle() local
H A Dgroup_norm.cpp229 at::Tensor input_reshaped = input.view({1, N * group, N ? -1 : 1}); in math_group_norm() local
H A Dlayer_norm.cpp235 at::Tensor input_reshaped = input.reshape({1, M, -1}); in math_native_layer_norm() local
H A DNormalization.cpp727 auto input_reshaped = input.contiguous().view_symint(shape); in instance_norm() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/linalg/
H A Dcholesky_op_gpu.cu.cc138 auto input_reshaped = input.template flat_inner_dims<Scalar, 3>(); in ComputeAsync() local
177 auto input_reshaped = const_cast<const Tensor*>(output) in ComputeAsync() local
H A Dmatrix_set_diag_op.cc160 auto input_reshaped = input.flat_inner_dims<T, 3>(); in Compute() local
H A Dmatrix_band_part_op.cc60 auto input_reshaped = input.flat_inner_dims<T, 3>(); in Compute() local
H A Dsvd_op_gpu.cu.cc115 auto input_reshaped = M_copy.template flat_inner_dims<Scalar, 3>(); in RunSVD() local
H A Dmatrix_diag_op.cc142 auto input_reshaped = input.flat_inner_dims<T, 3>(); in Compute() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DNormalization.cuh612 auto input_reshaped = input_.reshape({input_.size(0), input_.size(1), -1}); in batch_norm_backward_cuda_template() local
668 …auto input_reshaped = input_.reshape({input_.size(0), input_.size(1), -1}); // internally we merge… in batch_norm_stats_cuda_template() local
699 …auto input_reshaped = input_.reshape({input_.size(0), input_.size(1), -1}); // internally we merge… in batch_norm_elemt_cuda_template() local
788 …auto input_reshaped = input_.reshape({input_.size(0), input_.size(1), -1}); // internally we merge… in batch_norm_backward_reduce_cuda_template() local
844 …auto input_reshaped = input_.reshape({input_.size(0), input_.size(1), -1}); // internally we merge… in batch_norm_backward_elemt_cuda_template() local
895 …auto input_reshaped = input_.reshape({input_.size(0), input_.size(1), -1}); // internally we merge… in batch_norm_backward_elemt_cuda_template() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
H A Ddepthtospace_op.cc48 auto input_reshaped = NCHW_VECT_CToNCHW(input); in Compile() local
H A Dspacetodepth_op.cc48 auto input_reshaped = NCHW_VECT_CToNCHW(input); in Compile() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/batching_util/
H A Dconcat_split_util.h164 auto input_reshaped = in SplitCPU() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/
H A Dqlinear_dynamic.cpp510 auto input_reshaped = in apply_dynamic_impl() local