- Sort Score
- Result 10 results
- Languages All
Results 1 - 1 of 1 for output_batch_size (0.12 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/unroll_batch_matmul.cc
sliceInput(input_rhs, bcast.y_batch_size(), loc, rewriter); // Compute (single batch) MatMul for each output batch. std::vector<Value> matmuls; matmuls.reserve(bcast.output_batch_size()); for (int batch_idx : llvm::seq<int>(0, bcast.output_batch_size())) { int lhs_batch_idx, rhs_batch_idx; if (bcast.IsBroadcastingRequired()) { lhs_batch_idx = bcast.x_batch_indices()[batch_idx];
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 11.6K bytes - Viewed (0)