Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for getReductionIndices (0.33 sec)

  1. tensorflow/compiler/mlir/tensorflow/ir/tf_ops_a_m.cc

    // AllOp
    //===----------------------------------------------------------------------===//
    
    LogicalResult AllOp::verify() {
      AllOp op = *this;
      return VerifyReductionInputAndDims(op.getInput(), op.getReductionIndices(),
                                         op.getLoc());
    }
    
    //===----------------------------------------------------------------------===//
    // AnyOp
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 146.7K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tf2xla/transforms/legalize_tf.cc

        if (!input_ty) return failure();
        ArrayRef<int64_t> input_shape = input_ty.getShape();
    
        DenseIntElementsAttr dimensions;
        if (!matchPattern(op.getReductionIndices(), m_Constant(&dimensions)))
          return failure();
    
        // Build the final shape from input_shape and dimensions using a bitmap
        // to mark the reduced dimensions.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jun 11 20:00:43 UTC 2024
    - 291.8K bytes
    - Viewed (0)
Back to top