Searched defs:beta_data (Results 1 – 9 of 9) sorted by relevance
70 Tensor beta_data(DT_FLOAT, TensorShape({2})); in TestFoldOldBatchNorms() local154 Tensor beta_data(DT_FLOAT, TensorShape({4})); in TestFoldOldBatchNormsAfterDepthwiseConv2dNative() local232 Tensor beta_data(DT_FLOAT, TensorShape({2})); in TestFoldFusedBatchNorms() local309 Tensor beta_data(DT_FLOAT, TensorShape({4})); in TestFoldFusedBatchNormsAfterDepthwiseConv2dNative() local410 Tensor beta_data(DT_FLOAT, TensorShape({2})); in TestFoldFusedBatchNormsWithConcat() local502 Tensor beta_data(DT_FLOAT, TensorShape({2})); in TestFoldFusedBatchNormsWithBatchToSpace() local
36 float* beta_data) { in compute_fused_params()92 float* beta_data = beta.data_ptr<float>(); in q_batch_norm1d_impl() local201 float* beta_data = beta.data_ptr<float>(); in q_batch_norm2d_impl() local297 float* beta_data = beta.data_ptr<float>(); in q_batch_norm3d_impl() local
67 const CTYPE* beta_data; in layer_norm() local
87 scalar_t* beta_data = beta.data_ptr<scalar_t>(); in batch_norm_cpu_contiguous_impl() local139 scalar_t* beta_data = beta.data_ptr<scalar_t>(); in batch_norm_cpu_channels_last_impl() local724 opmath_t* beta_data = beta.data_ptr<opmath_t>(); in batch_norm_cpu_contiguous_impl() local787 opmath_t* beta_data = beta.data_ptr<opmath_t>(); in batch_norm_cpu_channels_last_impl() local
41 const T* beta_data = beta.defined() ? beta.const_data_ptr<T>() : nullptr; in LayerNormKernelImplInternal() local101 const param_t* beta_data = beta.defined() ? beta.const_data_ptr<param_t>() : nullptr; in layer_norm_kernel_mixed_type() local
48 const PT* beta_data = beta.defined() ? beta.const_data_ptr<PT>() : nullptr; in GroupNormKernelImplInternal() local303 const PT* beta_data = beta.defined() ? beta.const_data_ptr<PT>() : nullptr; in GroupNormKernelImplChannelsLastInternal() local
739 const T* beta_data, in launch_vectorized_layer_norm_kernel()773 const T* beta_data = beta.defined() ? beta.const_data_ptr<T>() : nullptr; in LayerNormKernelImplInternal() local
609 const T* beta_data = beta.defined() ? beta.const_data_ptr<T>() : nullptr; in GroupNormKernelImplInternal() local
2786 const float* beta_data = beta.defined() ? beta.const_data_ptr<float>() : nullptr; local3074 const float* beta_data = beta.defined() ? beta.const_data_ptr<float>() : nullptr; local