- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for NormalizeIndexVector (0.15 sec)
-
tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_hlo_conversions/scatter.h
// window dims are the trailing dimensions. // // To canonicalize the updates above, replace the updates with: // transpose(updates, permutation={3,4,0,1,2}) // // Note: NormalizeIndexVector is assumed to have run on the indices already so // that the index_vector_dim is the trailing dimension in `indices`. LogicalResult CanonicalizeScatterUpdates(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 10.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_hlo_conversions/util.cc
GetInversePermutationArray(permutation_array); return GetPermutationAndTransposedShape(inverse_permutation_array, input_type, rewriter); } LogicalResult NormalizeIndexVector(Operation* parent_op, Value& indices, ShapedType& indices_type, int64_t index_vector_dim,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 10.1K bytes - Viewed (0)