Home | History | Annotate | Download | only in kernels

Lines Matching refs:channel_count

44     int64 channel_count = 0;
78 const int64 channel_count = input.NumElements() / channels;
85 options.channel_count = channel_count;
104 const int64 channel_count = options.channel_count;
105 auto input_data = input->shaped<float, 2>({channel_count, kChannelSize});
109 auto output_data = output->shaped<float, 2>({channel_count, kChannelSize});
116 Shard(worker_threads.num_threads, worker_threads.workers, channel_count,
158 functor::AdjustHsvInYiqGPU()(ctx, options.channel_count, options.input,