Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 4 of 4 for NormalizeIndexVector (0.5 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.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)
  3. 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)
  4. 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)
Back to top