Home
last modified time | relevance | path

Searched defs:dx_ptr (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/
H A Dbatch_norm_kernel.cpp479 scalar_t* dx_ptr = grad_input_data + n * n_channel * image_size + c * image_size; in batch_norm_cpu_backward_contiguous_impl() local
499 scalar_t* dx_ptr = grad_input_data + n * n_channel * image_size + c * image_size; in batch_norm_cpu_backward_contiguous_impl() local
635 scalar_t* dx_ptr = grad_input_data + i * n_channel; in batch_norm_cpu_backward_channels_last_impl() local
1073 scalar_t* dx_ptr = grad_input_data + n * n_channel * image_size + c * image_size; in batch_norm_cpu_backward_contiguous_internal() local
1084 scalar_t* dx_ptr = grad_input_data + n * n_channel * image_size + c * image_size; in batch_norm_cpu_backward_contiguous_internal() local
1231 scalar_t* dx_ptr = grad_input_data + i * n_channel; in batch_norm_cpu_backward_channels_last_internal() local
/aosp_15_r20/external/tensorflow/tensorflow/tools/android/test/jni/object_tracking/
H A Dobject_tracker.cc429 const int32_t* dx_ptr = dx[y]; in Draw() local