| resampler_ops.cc | 135 // tensor of dimension [batch, dim_0, ... dim_n, 4, data_channels].
139 int64 data_channels, int warp_dims) {
143 // data_channels], the offset dimensions for Gather is the last 3 dimensions.
157 // Output dimensions are [batch, dim_0, ... dim_n, 2, 2, data_channels].
159 /*slice_sizes=*/{1, 2, 2, data_channels});
165 // resulting tensor of dimension: [batch, dim_0, ...dim_n, 2, 2, data_channels].
173 // data_channels], the update window dimensions is the last 3 dimensions.
251 int64 last_warp_dim, int64 data_channels,
271 weights_with_channels_dims.push_back(data_channels);
327 /*limit_indices=*/{batch_size, width + 1, height + 1, data_channels},
492 const int64 data_channels = data_shape.dim_size(3); variable
613 const int64 data_channels = data_shape_tf.dim_size(3); variable
[all...] |
