- Sort Score
- Result 10 results
- Languages All
Results 1 - 7 of 7 for CastOp (0.66 sec)
-
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) -
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) -
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) -
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) -
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) -
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) -
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)