Lines Matching refs:in_depth
30 int in_depth;
47 in_depth(0),
112 // filter_inner_dim_size = in_depth * depth_multiplier
114 // [filter_rows, filter_cols, in_depth, depth_multiplier]
118 // in_depth = 3, depth_multiplier = 2, filter [2, 2], register_width = 4
121 // filter [rows, cols, in_depth, depth_multiplier]
125 // padded_filter [rows, cols, in_depth, depth_multiplier]
175 // in_depth = 3, depth_multiplier = 2, filter [2, 2], register_width = 4
177 // input: [batch, in_rows, in_cols, in_depth]
198 // Calculate vectorized and scalar (residual) lengths for 'in_depth'.
200 (args.in_depth / kPacketSize) * kPacketSize;
201 const int64 input_scalar_size = args.in_depth % kPacketSize;
217 // Iterate through all rows x cols reading 'in_depth' from 'input' and
232 auto* in = input + (in_r * args.in_cols + in_c) * args.in_depth;