- Sort Score
- Result 10 results
- Languages All
Results 41 - 50 of 81 for is_constant (0.54 sec)
-
tensorflow/compiler/jit/clone_constants_for_better_clustering.cc
return total_elements < kSmallTensorThreshold; } // We only clone small constants since we want to avoid increasing memory // pressure on GPUs. absl::StatusOr<bool> IsSmallConstant(Node* n) { if (!n->IsConstant()) { return false; } return IsConstantSmall(n); } bool IsInPlaceOp(absl::string_view op_name) { return op_name == "InplaceUpdate" || op_name == "InplaceAdd" ||
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 12 06:33:33 UTC 2024 - 7.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/experimental/tac/transforms/device_transform_patterns.cc
mlir::DenseIntElementsAttr splits_attr; if (!splits || !matchPattern(splits, m_Constant(&splits_attr))) return failure(); auto split_dim = splitv_op.getSplitDim().getDefiningOp(); mlir::ElementsAttr split_dim_attr; if (!split_dim || !matchPattern(split_dim, m_Constant(&split_dim_attr))) return failure(); auto input = splitv_op.getValue();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 25.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/transforms/legalize_tf.cc
m_Constant(&window_dimensions)) && matchPattern(op.getWindowStrides(), m_Constant(&window_strides)) && matchPattern(op.getBaseDilations(), m_Constant(&base_dilations)) && matchPattern(op.getWindowDilations(), m_Constant(&window_dilations)) && matchPattern(op.getPadding(), m_Constant(&padding)))) return failure();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 20:00:43 UTC 2024 - 291.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/ir/tfl_ops.cc
// Get the tensors. DenseElementsAttr input_tensor, weights_tensor, bias_tensor; if (!matchPattern(getInput(), m_Constant(&input_tensor)) || !matchPattern(getFilter(), m_Constant(&weights_tensor)) || (has_bias && !matchPattern(getBias(), m_Constant(&bias_tensor)))) { return failure(); } // Get the tensor types.
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/transforms/shape_inference.cc
matchPattern(op.getWindowDimensions(), m_Constant(&window_dimensions)) && matchPattern(op.getWindowStrides(), m_Constant(&window_strides)) && matchPattern(op.getBaseDilations(), m_Constant(&base_dilations)) && matchPattern(op.getWindowDilations(), m_Constant(&window_dilations)) && matchPattern(op.getPadding(), m_Constant(&padding))) {
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/quantization/ir/ConvertConst.cc
LogicalResult QuantizedConstRewrite::matchAndRewrite( QuantizeCastOp qbarrier, PatternRewriter &rewriter) const { Attribute value; // Is the operand a constant? if (!matchPattern(qbarrier.getArg(), m_Constant(&value))) { return failure(); } // Does the qbarrier convert to a quantized type. This will not be true // if a quantized type has not yet been chosen or if the cast to an equivalent
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 4.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_ops_a_m.cc
SmallVectorImpl<int64_t>& s0_shape, SmallVectorImpl<int64_t>& s1_shape) { if (!matchPattern(op.getS0(), m_Constant(&s0))) return false; if (!matchPattern(op.getS1(), m_Constant(&s1))) return false; for (auto s : s0.getValues<APInt>()) s0_shape.push_back(s.getSExtValue()); for (auto s : s1.getValues<APInt>()) s1_shape.push_back(s.getSExtValue());
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/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/quantization/stablehlo/passes/post_quantize.cc
LogicalResult matchAndRewrite(mlir::stablehlo::UniformQuantizeOp op, PatternRewriter& rewriter) const override { DenseFPElementsAttr attr; if (matchPattern(op.getOperand(), m_Constant(&attr))) { const Type qtype = op.getResult().getType(); ElementsAttr quantized_attr = Quantize(attr, qtype); if (quantized_attr) { rewriter.replaceOpWithNewOp<mlir::stablehlo::ConstantOp>(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 05 07:39:40 UTC 2024 - 6.3K 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)