- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for sizeType (0.13 sec)
-
tensorflow/compiler/mlir/lite/ir/tfl_ops.cc
slice_op.setOperand(1, new_begin); return success(); } } // Handle size. if (size_op && size_type && size_type.getElementType().isInteger(64)) { auto new_size = NarrowDownInt64InputValuesForOp( size_op, size_type, slice_op.getLoc(), &rewriter); if (new_size != nullptr) { slice_op.setOperand(2, new_size); return success(); }
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/lite/stablehlo/transforms/uniform_quantized_stablehlo_to_tfl_pass.cc
lower_limit_cst); SmallVector<int64_t> size_len{rank}; TensorType size_type = operand_type.cloneWith(size_len, i64_type); auto size_attr = DenseIntElementsAttr::get(size_type, slice_sizes); auto size = rewriter.create<arith::ConstantOp>(op.getLoc(), size_attr); rewriter.replaceOpWithNewOp<TFL::SliceOp>(op, output, input, begin, size);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 22 09:00:19 UTC 2024 - 99.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/shape_inference.cc
func::FuncOp ret = queue_.front(); queue_.pop(); queue_set_.erase(ret); return ret; } // Returns the current size of the queue. std::queue<func::FuncOp>::size_type QueueSize() const { return queue_.size(); } Dialect* const tf_dialect_; private: // Returns whether the result of an operation could be updated to a new
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Jun 08 07:28:49 UTC 2024 - 134.1K bytes - Viewed (0)