- Sort Score
- Result 10 results
- Languages All
Results 1 - 4 of 4 for operand_shape (0.51 sec)
-
tensorflow/compiler/mlir/lite/stablehlo/transforms/uniform_quantized_stablehlo_to_tfl_pass.cc
// Shape after dilation. SmallVector<int64_t> dilated_shape(rank); ArrayRef<int64_t> operand_shape = operand_type.getShape(); for (int i = 0; i < rank; ++i) { dilated_shape[i] = operand_shape[i] + interior_padding_i64[i] * (operand_shape[i] - 1); } TensorType output_type = op.getResult().getType().cast<TensorType>();
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/lite/stablehlo/transforms/legalize_hlo.cc
// Calculates the flattened shapes for dynamic shaped operands in // mhlo.dot_general: // 1. flattened_out_dim = UnsortedSegmentProdOp(operand_shape, out_axes) // 2. flattened_contracting_dim = UnsortedSegmentProdOp(operand_shape, // contracting_axes) // 3. batch_dimensions = Gather(operand_shape, batch_axes) // 4. flattened_shape = Concat(batch_dimensions, flattened_out_dim, // flattened_contracting_dim)
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/transforms/shape_inference.cc
XlaSelectAndScatterOp op) { DCOMMENT_OP(op, "Inferring shape for XlaSelectAndScatterOp"); auto operand_shape = mlir::cast<ShapedType>(op.getOperand().getType()); auto source_shape = mlir::cast<ShapedType>(op.getSource().getType()); DenseElementsAttr window_dimensions, window_strides, padding; if (operand_shape.hasRank() && source_shape.hasRank() && matchPattern(op.getWindowDimensions(), m_Constant(&window_dimensions)) &&
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/ir/tfl_ops.cc
if (shaped_type.hasStaticShape()) { has_static_operand_shape = true; operand_shape = shaped_type.getShape(); } } SmallVector<int64_t, 4> broadcastedShape; if (has_static_cond_shape && has_static_operand_shape && !OpTrait::util::getBroadcastedShape(cond_shape, operand_shape, broadcastedShape)) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 09:41:17 UTC 2024 - 169.2K bytes - Viewed (0)