- Sort Score
- Result 10 results
- Languages All
Results 1 - 1 of 1 for batch_dimensions (0.2 sec)
-
tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_hlo.cc
// contracting_axes) // 3. batch_dimensions = Gather(operand_shape, batch_axes) // 4. flattened_shape = Concat(batch_dimensions, flattened_out_dim, // flattened_contracting_dim) // The flattened shape for LHS // is like [batch_dimensions, flattened_out_dimension, // flattened_contracting_dimension] and [batch_dimensions, // flattened_contracting_dimension, flattened_out_dimension] for RHS.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 154.9K bytes - Viewed (0)