Searched defs:gamma_data (Results 1 – 7 of 7) sorted by relevance
75 Tensor gamma_data(DT_FLOAT, TensorShape({2})); in TestFoldOldBatchNorms() local159 Tensor gamma_data(DT_FLOAT, TensorShape({4})); in TestFoldOldBatchNormsAfterDepthwiseConv2dNative() local237 Tensor gamma_data(DT_FLOAT, TensorShape({2})); in TestFoldFusedBatchNorms() local314 Tensor gamma_data(DT_FLOAT, TensorShape({4})); in TestFoldFusedBatchNormsAfterDepthwiseConv2dNative() local415 Tensor gamma_data(DT_FLOAT, TensorShape({2})); in TestFoldFusedBatchNormsWithConcat() local507 Tensor gamma_data(DT_FLOAT, TensorShape({2})); in TestFoldFusedBatchNormsWithBatchToSpace() local
40 const T* gamma_data = gamma.defined() ? gamma.const_data_ptr<T>() : nullptr; in LayerNormKernelImplInternal() local100 const param_t* gamma_data = gamma.defined() ? gamma.const_data_ptr<param_t>() : nullptr; in layer_norm_kernel_mixed_type() local190 const T2* gamma_data, in layer_norm_backward_frame()305 const float* gamma_data, in layer_norm_backward_frame()512 const T2* gamma_data = in LayerNormBackwardKernelImplInternal() local
47 const PT* gamma_data = gamma.defined() ? gamma.const_data_ptr<PT>() : nullptr; in GroupNormKernelImplInternal() local302 const PT* gamma_data = gamma.defined() ? gamma.const_data_ptr<PT>() : nullptr; in GroupNormKernelImplChannelsLastInternal() local904 const PT* gamma_data = gamma.defined() ? gamma.const_data_ptr<PT>() : nullptr; in GroupNormBackwardKernelImplInternal() local1384 const PT* gamma_data = gamma.defined() ? gamma.const_data_ptr<PT>() : nullptr; in GroupNormBackwardKernelImplChannelsLastInternal() local
61 const CTYPE* gamma_data; in layer_norm() local
608 const T* gamma_data = gamma.defined() ? gamma.const_data_ptr<T>() : nullptr; in GroupNormKernelImplInternal() local692 const T* gamma_data = gamma.defined() ? gamma.const_data_ptr<T>() : nullptr; in GroupNorm1dBackward() local833 const T* gamma_data = gamma.defined() ? gamma.const_data_ptr<T>() : nullptr; in GroupNormBackwardKernelImplInternal() local
738 const T* gamma_data, in launch_vectorized_layer_norm_kernel()772 const T* gamma_data = gamma.defined() ? gamma.const_data_ptr<T>() : nullptr; in LayerNormKernelImplInternal() local1156 const T* gamma_data = in LayerNormBackwardKernelImplInternal() local
2785 const float* gamma_data = gamma.defined() ? gamma.const_data_ptr<float>() : nullptr; local3073 const float* gamma_data = gamma.defined() ? gamma.const_data_ptr<float>() : nullptr; local