Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for NormalizeIndexVector (0.15 sec)

  1. 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)
  2. 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)
Back to top