- Sort Score
- Result 10 results
- Languages All
Results 1 - 5 of 5 for m_Constant (0.27 sec)
-
tensorflow/compiler/mlir/lite/transforms/prepare_tf.cc
if (!begin_attr_type || !matchPattern(begin_attr, m_Constant(&begin_elem_attr))) { return failure(); } if (!end_attr_type || !matchPattern(end_attr, m_Constant(&end_elem_attr))) { return failure(); } if (!strides_attr_type || !matchPattern(strides_attr, m_Constant(&strides_elem_attr))) { return failure(); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 28 21:49:50 UTC 2024 - 64.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/lower_tf.cc
ElementsAttr paddings; llvm::SmallVector<int64_t, 4> block_shape_ints; auto padded_shape = llvm::to_vector<4>(input_shape); if (matchPattern(op.getBlockShape(), m_Constant(&block_shape)) && matchPattern(op.getPaddings(), m_Constant(&paddings))) { for (uint64_t i = 0; i < block_rank; i++) { int64_t paddings_sum = paddings.getValues<APInt>()[{i, 0}].getSExtValue() +
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 74.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/lower_static_tensor_list.cc
: op.emitOpError(error_info); } } } DenseIntElementsAttr dense_elem_attr; if (matchPattern(element_shape, m_Constant(&dense_elem_attr))) { // Note: It's technically unsafe to rewrite // TensorListReserve(num_element, element_shape) // to // Fill(Concat(num_element, element_shape), 0)
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 20:00:43 UTC 2024 - 70.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/quantize_composite_functions.cc
private: LogicalResult matchAndRewrite(quantfork::QuantizeCastOp q_op, PatternRewriter& rewriter) const override { DenseFPElementsAttr attr; if (!matchPattern(q_op.getArg(), m_Constant(&attr))) { return failure(); } ShapedType tensor_qtype = mlir::cast<ShapedType>(q_op.getResult().getType()); Attribute tensor_proto_attr = Quantize(attr, tensor_qtype);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 54.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/flatbuffer_import.cc
} if (op_name == "tfl.reshape") { // Flattens reshape ops when more than one dimension shape operand is given. mlir::DenseIntElementsAttr shape_attr; if (matchPattern(op_state.operands[1], m_Constant(&shape_attr))) { auto shape_ty = op_state.operands[1].getType().dyn_cast<RankedTensorType>(); if (shape_ty != nullptr && shape_ty.hasRank() && shape_ty.getRank() > 1) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 21 18:21:50 UTC 2024 - 66.8K bytes - Viewed (0)