Searched refs:tf_kernel (Results 1 – 1 of 1) sorted by relevance
3455 const auto tf_kernel = attrs.get<std::vector<int64>>("ksize"); in ConvertPool3D() local3456 if (tf_kernel.size() != kNumDims) { in ConvertPool3D()3460 if (tf_kernel[0] != 1 || tf_kernel[c_index] != 1) { in ConvertPool3D()3476 const nvinfer1::Dims3 ksize(tf_kernel[d_index], tf_kernel[h_index], in ConvertPool3D()3477 tf_kernel[w_index]); in ConvertPool3D()3714 const auto tf_kernel = attrs.get<std::vector<int64>>("ksize"); in ConvertPool() local3715 const nvinfer1::DimsHW ksize(tf_kernel[h_index], tf_kernel[w_index]); in ConvertPool()