Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 1 of 1 for ReplaceOpWithNewOp (0.33 sec)

  1. tensorflow/compiler/mlir/tf2xla/transforms/legalize_tf.cc

        // implemented.
        if (params_ty.hasStaticShape() && indices_ty.hasStaticShape()) {
          rewriter.replaceOpWithNewOp<mhlo::GatherOp>(
              op, op.getType(), op.getParams(), op.getIndices(), dims_attr,
              GetI64ElementsAttr(slice_sizes, &rewriter));
        } else {
          rewriter.replaceOpWithNewOp<mhlo::DynamicGatherOp>(
              op, op.getType(), op.getParams(), op.getIndices(), slice_sizes_value,
    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