- Sort Score
- Result 10 results
- Languages All
Results 1 - 4 of 4 for NormalizeIndexVector (0.5 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.h
// If index_vector_dim == indices.rank() then insert the implicit extra // dimension into indices to normalize everything to index_vector_dim == // indices.rank() - 1. 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: Wed Nov 08 11:35:25 UTC 2023 - 6.4K 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) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_hlo.cc
} // Normalize start_indices so index_vector_dim == start_indices.rank() - 1. int64_t index_vector_dim = gather_op.getDimensionNumbers().getIndexVectorDim(); if (failed(NormalizeIndexVector(gather_op, start_indices, start_indices_type, index_vector_dim, rewriter))) { return failure(); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 154.9K bytes - Viewed (0)