Home
last modified time | relevance | path

Searched refs:kernel_sizes_tensor (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/core/framework/
Dcommon_shape_fns.cc973 const Tensor* kernel_sizes_tensor = c->input_tensor(c->num_inputs() - 2); in MaxPoolV2Shape() local
974 if (kernel_sizes_tensor == nullptr) { in MaxPoolV2Shape()
978 kernel_sizes.resize(kernel_sizes_tensor->shape().num_elements()); in MaxPoolV2Shape()
979 auto kernel_sizes_vec = kernel_sizes_tensor->flat<int32>(); in MaxPoolV2Shape()