HomeSort by relevance Sort by last modified time
    Searched refs:mean_ptr (Results 1 - 2 of 2) sorted by null

  /external/tensorflow/tensorflow/core/kernels/
quantized_instance_norm.cc 139 const float* mean_ptr, const float* variance_ptr,
146 const float32x4_t mean[4] = {vld1q_f32(mean_ptr + col_offset),
147 vld1q_f32(mean_ptr + col_offset + 4),
148 vld1q_f32(mean_ptr + col_offset + 8),
149 vld1q_f32(mean_ptr + col_offset + 12)};
195 const uint32_t cols, const float* mean_ptr,
203 const float32x4_t mean[4] = {vld1q_f32(mean_ptr + col_offset + 12),
204 vld1q_f32(mean_ptr + col_offset + 8),
205 vld1q_f32(mean_ptr + col_offset + 4),
206 vld1q_f32(mean_ptr + col_offset)}
    [all...]
fused_batch_norm_op.cc 475 auto mean_ptr = StreamExecutorUtil::AsDeviceMemory<U>(mean); local
487 y_backprop_ptr, x_ptr, scale_ptr, mean_ptr, inv_variance_ptr,
    [all...]

Completed in 88 milliseconds