Searched refs:paddings_dims (Results 1 – 2 of 2) sorted by relevance
54 const std::vector<int>& paddings_dims = paddings_array.shape().dims(); in Run() local55 if (paddings_dims.size() != 2) { in Run()62 for (int i = 0; i < paddings_dims[0]; ++i) { in Run()
1923 const Dims<4>& paddings_dims, T* output_data, in SpaceToBatchND() argument1931 DimsToShape(paddings_dims), paddings_data, in SpaceToBatchND()1940 const Dims<4>& paddings_dims, T* output_data, in SpaceToBatchND() argument1947 DimsToShape(paddings_dims), paddings_data, in SpaceToBatchND()