Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 7 of 7 for CastOp (0.18 sec)

  1. tensorflow/compiler/mlir/tensorflow/transforms/shape_inference.cc

      while (!worklist.empty()) {
        mlir::Operation* const use = worklist.pop_back_val();
    
        // Follow the `CastOp`/`IdentityOp`'s users to handle the `RestoreV2` ->
        // (optionally `IdentityOp`) -> `CastOp` `AssignVariableOp` case.
        if (llvm::isa<TF::CastOp, TF::IdentityOp>(use)) {
          llvm::append_range(worklist, use->getUsers());
          continue;
        }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Sat Jun 08 07:28:49 UTC 2024
    - 134.1K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_hlo.cc

            RankedTensorType::get(transpose_params.canonicalized_output_shape,
                                  result_type.getElementType());
    
        TF::CastOp cast_op = nullptr;
        if (start_indices_type.getElementType().isUnsignedInteger(32)) {
          cast_op = rewriter.create<TF::CastOp>(
              gather_op->getLoc(),
              RankedTensorType::get(start_indices_type.getShape(),
                                    rewriter.getI64Type()),
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 154.9K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/tensorflow/ir/tf_ops_a_m.cc

    }
    
    //===----------------------------------------------------------------------===//
    // CastOp
    //===----------------------------------------------------------------------===//
    
    OpFoldResult CastOp::fold(FoldAdaptor) {
      // Cast with the same type is a no-op.
      Value operand = getOperand();
      if (getType() == operand.getType()) return operand;
      return {};
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 146.7K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/lite/ir/tfl_ops.cc

      results.add<FoldPseudoConstOp>(context);
    }
    
    //===----------------------------------------------------------------------===//
    // CastOp
    //===----------------------------------------------------------------------===//
    
    OpFoldResult CastOp::fold(FoldAdaptor adaptor) {
      auto operands = adaptor.getOperands();
      assert(operands.size() == 1);
      if (getInput().getType() == getType()) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 02 09:41:17 UTC 2024
    - 169.2K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/tensorflow/ir/tf_ops_n_z.cc

          if (body_arg.getType() == cond_arg.getType() &&
              body_arg.getType() != while_operand.getType()) {
            changed = true;
            rewriter.setInsertionPoint(while_op);
            auto cast_op = rewriter.create<CastOp>(
                while_op.getLoc(), body_arg.getType(), while_operand);
            while_op.setOperand(op_idx, cast_op);
          }
        }
        return success(changed);
      }
    };
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 09 22:07:10 UTC 2024
    - 170.8K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/tensorflow/translate/import_model.cc

                mlir::tf_saved_model::GetBoundInputArgTypeFor(global_tensor);
            arg.setType(new_type);
            if (global_tensor.getIsMutable()) {
              auto arg_with_original_type = builder.create<mlir::TF::CastOp>(
                  global_tensor.getLoc(), old_type, arg,
                  /*Truncate=*/builder.getBoolAttr(false));
              arg.replaceAllUsesWith(arg_with_original_type);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 01 11:17:36 UTC 2024
    - 183.2K bytes
    - Viewed (0)
  7. src/cmd/internal/obj/arm64/asm7.go

    	{ASTP, C_PAIR, C_NONE, C_NONE, C_NPOREG, C_NONE, 67, 4, 0, 0, 0},
    	{ASTP, C_PAIR, C_NONE, C_NONE, C_NPOREG, C_NONE, 67, 4, 0, 0, C_XPRE},
    	{ASTP, C_PAIR, C_NONE, C_NONE, C_NPOREG, C_NONE, 67, 4, 0, 0, C_XPOST},
    	{ASTP, C_PAIR, C_NONE, C_NONE, C_PPOREG, C_NONE, 67, 4, 0, 0, 0},
    	{ASTP, C_PAIR, C_NONE, C_NONE, C_PPOREG, C_NONE, 67, 4, 0, 0, C_XPRE},
    	{ASTP, C_PAIR, C_NONE, C_NONE, C_PPOREG, C_NONE, 67, 4, 0, 0, C_XPOST},
    Registered: Wed Jun 12 16:32:35 UTC 2024
    - Last Modified: Wed May 15 15:44:14 UTC 2024
    - 201.1K bytes
    - Viewed (0)
Back to top