- Sort Score
- Result 10 results
- Languages All
Results 11 - 20 of 230 for GetOperation (0.17 sec)
-
tensorflow/compiler/mlir/lite/transforms/decompose_hybrid_quantization.cc
public: using OpRewritePattern<SrcOp>::OpRewritePattern; LogicalResult matchAndRewrite(SrcOp srcop, PatternRewriter &rewriter) const final { Operation *op = srcop.getOperation(); bool allTypesFp = true; bool allTypesQuantizedOrInt = true; for (auto operand : op->getOperands()) { ShapedType type = mlir::dyn_cast<ShapedType>(operand.getType()); if (!type) continue;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 5.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/quantization/tensorflow/tf_to_quant.cc
} // Use the min/max from the operands and the num_bits and narrow_range // attribute to create the quantization parameter for the new quantize op. rewriter.setInsertionPointAfter(tf_op.getOperation()); IntegerAttr num_bits = rewriter.getI64IntegerAttr(tf_op.getNumBits()); BoolAttr narrow_range = rewriter.getBoolAttr(tf_op.getNarrowRange()); Type res_type = tf_op.getType();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 8.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_executor.cc
<< ")"; } } return success(); } void GraphOp::print(OpAsmPrinter &p) { p << ' '; p.printRegion(getOperation()->getRegion(0)); p.printOptionalAttrDict(getOperation()->getAttrs()); } ParseResult GraphOp::parse(OpAsmParser &parser, OperationState &result) { llvm::SMLoc loc = parser.getCurrentLocation(); // Parse the body region.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 42.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/layout_optimization.cc
operand->get().getType() == ShuffleRankedTensorType(ranked_tranpose_type, permutation)) { TransposeOp transpose = tranpose_op; transpose.getOperation()->moveBefore(op); transpose.setOperand(0, operand->get()); transpose.setOperand(1, permutation_op); transpose_ops->erase(it); return transpose; } } }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 19.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tpu_partitioned_op_conversion.cc
llvm::SmallVector<TF::TPUReplicateMetadataOp, 4> metadata; getOperation()->walk( [&metadata](TF::TPUReplicateMetadataOp op) { metadata.push_back(op); }); IntegerAttr num_cores_per_replica; if (metadata.size() == 1) { num_cores_per_replica = metadata.front().getNumCoresPerReplicaAttr(); } auto result = getOperation()->walk([&num_cores_per_replica](Operation* op) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 5.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/composite_lowering_pass.cc
target.addLegalDialect<TFL::TensorFlowLiteDialect>(); target.addLegalDialect<arith::ArithDialect>(); if (failed(applyPartialConversion(getOperation(), target, std::move(patterns)))) { getOperation().emitError("Composite lowering pass failed."); signalPassFailure(); } } } // namespace // Creates an instance of the pass.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 28 23:16:05 UTC 2024 - 3K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/post_quantize.cc
LogicalResult matchAndRewrite(OpTy op, PatternRewriter& rewriter) const override { if (op.getOperation()->template hasTrait<OpTrait::IsTerminator>()) { return failure(); } for (auto result : op.getOperation()->getOpResults()) { if (!result.use_empty()) { return failure(); } }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 17.1K bytes - Viewed (0) -
tests/integration/telemetry/api/testdata/attributegen.yaml
Registered: Fri Jun 14 15:00:06 UTC 2024 - Last Modified: Tue Mar 12 00:53:57 UTC 2024 - 639 bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/executor_tpuv1_island_coarsening.cc
} } } void TpuV1BridgeExecutorIslandCoarsening::runOnOperation() { SymbolTable symbol_table(getOperation()); // Map tpu cluster names to the functions that contain operations for this // cluster. DenseMap<StringRef, DenseSet<func::FuncOp>> tpu_funcs; for (func::FuncOp func_op : getOperation().getOps<func::FuncOp>()) { func_op.walk([&](Operation* op) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 17:58:54 UTC 2024 - 27.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/extract_tpu_copy_with_dynamic_shape_op.cc
.getOperation(); tpu_copy_with_dynamic_shape_op->moveBefore(return_op); return success(); } // Update all the usage of tf_device.return op with launch op result. void UpdateReturnOpResultWithLaunchOpResult(tf_device::LaunchOp* launch_op) { auto operand_not_in_launch = [&](OpOperand& operand) { return !launch_op->getOperation()->isProperAncestor(operand.getOwner()); };
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 8.6K bytes - Viewed (0)