Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 4 of 4 for SmallVectorImpl (0.38 sec)

  1. tensorflow/compiler/mlir/lite/transforms/lower_static_tensor_list.cc

    void UpdateTensorListTypes(
        const llvm::SmallSet<int, 4> &tensor_list_index,
        const llvm::SmallSet<int, 4> &resized_tensor_list_index,
        ArrayRef<Type> types, R &&range, ValueRange operands,
        llvm::SmallVectorImpl<Type> *updated_types) {
      int i = 0;
      for (const auto it : llvm::zip(types, range, operands)) {
        if (tensor_list_index.count(i)) {
          // Only change the tensorlist's type to unranked tensor if it has been
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jun 11 20:00:43 UTC 2024
    - 70.7K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tf2xla/transforms/legalize_tf.cc

      return GetTensorFeatureDimIndex(input_ty.getRank(), format);
    }
    
    // Gets all integer values from the given attribute and push them to `values`.
    void GetI64ArrayAttrValues(Attribute attr, SmallVectorImpl<int64_t> *values) {
      auto array_attr = mlir::cast<ArrayAttr>(attr);
      values->reserve(array_attr.getValue().size());
      for (Attribute val : array_attr.getValue())
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jun 11 20:00:43 UTC 2024
    - 291.8K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/tensorflow/ir/tf_generated_ops.td

        // values, respectively.
        bool GetSlicedShapeAndBoundRanges(
          ::llvm::SmallVectorImpl<int64_t> *input_shape,
          ::llvm::SmallVectorImpl<int64_t> *slice_begin,
          ::llvm::SmallVectorImpl<int64_t> *slice_end,
          ::llvm::SmallVectorImpl<int64_t> *slice_stride);
      }];
    }
    
    def TF_StringJoinOp : TF_Op<"StringJoin", [Pure]> {
      let summary = [{
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jun 11 23:24:08 UTC 2024
    - 793K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/tensorflow/transforms/shape_inference.cc

        absl::flat_hash_map<ValuePort, Attribute, ValuePortHasher>;
    using ComputedQueryFn = function_ref<bool(ValuePort)>;
    using ValueQueryFn = function_ref<Attribute(const ValuePort&)>;
    using ValuePortInputs = SmallVectorImpl<ValuePort>;
    
    // Note: Following implements the rank 1 pack op case so could be
    // generalized.
    //
    // Maps the specified component in the `port` of the given op's result to one of
    // the element in the input.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Sat Jun 08 07:28:49 UTC 2024
    - 134.1K bytes
    - Viewed (0)
Back to top