Searched defs:data_channels (Results 1 – 3 of 3) sorted by relevance
142 int64_t data_channels, int warp_dims) { in Gather2by2Neighbors()254 int64_t last_warp_dim, int64_t data_channels, in CalculateGradData()364 TensorShape warp_shape, int64_t data_channels, in CalculateGradWarp()494 const int64_t data_channels = data_shape.dim_size(3); in Compile() local611 const int64_t data_channels = data_shape_tf.dim_size(3); in Compile() local
443 std::vector<rtc::scoped_refptr<DataChannelInterface>>& data_channels() { in data_channels() function
821 nir_def *data_channels[NIR_MAX_VEC_COMPONENTS]; in vectorize_stores() local