Searched refs:batch_dim_size (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | image_resize_ops.cc | 297 const int64 batch_dim_size = in ResizeUsingDilationAndConvolution() local
302 {batch_dim_size, in_size[0], in_size[1], channels}, {1, 1, 1, 1}); in ResizeUsingDilationAndConvolution()
311 {batch_dim_size, in_size[0] + num_extended[0], in_size[1], channels}, in ResizeUsingDilationAndConvolution()
|