- Sort Score
- Result 10 results
- Languages All
Results 1 - 1 of 1 for batch_dim_size (0.17 sec)
-
tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_hlo.cc
// where the permutation is [0, 2, 3, 1, 4] int batch_idx = 0; int offset_idx = 0; int batch_dim_size = batch_dims.size(); for (int i = 0; i < output_rank; ++i) { if (batch_idx >= batch_dims.size()) { transpose_params.permutation.push_back(batch_dim_size + offset_idx); ++offset_idx; } else if (offset_idx < original_offset_dims.size() &&
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 154.9K bytes - Viewed (0)