Searched defs:out_row (Results 1 – 3 of 3) sorted by relevance
95 const int out_row = (thread_id / out_depth / out_width) % out_height; in DepthwiseConv2dGPUKernelNHWC() local336 const int out_row = (thread_id / out_width) % out_height; in DepthwiseConv2dGPUKernelNCHW() local799 NOUNROLL for (int out_row = out_row_start; out_row <= out_row_end; in DepthwiseConv2dBackpropInputGPUKernelNHWC() local875 UNROLL for (int out_row = out_row_start; out_row <= out_row_end; in DepthwiseConv2dBackpropInputGPUKernelNCHW() local1011 const int out_row = (thread_id / out_depth / out_width) % out_height; in DepthwiseConv2dBackpropFilterGPUKernelNHWC() local1290 const int out_row = (thread_id / out_width) % out_height; in DepthwiseConv2dBackpropFilterGPUKernelNCHW() local
174 float* out_row = out + (y * wanted_width * wanted_channels); in ReadTensorFromImageFile() local
363 float* out_row = out + (y * wanted_width * wanted_channels); in TensorFromFrame() local