- Sort Score
- Result 10 results
- Languages All
Results 1 - 1 of 1 for GetConv2DBlockSize (0.26 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/tpu_space_to_depth_pass.cc
} } } // Gets block size that is equal to stride from spatial dimension // from convolution. // Space to depth transform won't be triggered if block size <= 1. int32_t GetConv2DBlockSize(TF::Conv2DOp conv2d) { SmallVector<int32_t, 4> strides(4, 1); for (int i = 0; i < 3; ++i) { strides[i] = mlir::cast<mlir::IntegerAttr>(conv2d.getStrides()[i]).getInt(); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 29.3K bytes - Viewed (0)