- Sort Score
- Result 10 results
- Languages All
Results 1 - 6 of 6 for GetBoolAttr (0.38 sec)
-
tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_hlo.cc
Value output = rewriter.create<TF::ResizeBilinearOp>( conv_op.getLoc(), conv_op.getType(), conv_op.getLhs(), output_sizes_attr, /*align_corners=*/rewriter.getBoolAttr(align_corners), /*half_pixel_centers=*/rewriter.getBoolAttr(false)); rewriter.replaceOp(conv_op, {output}); } LogicalResult MatchToResizeBilinearOp( mhlo::ConvolutionOp conv_op, bool& align_corners,
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/lite/transforms/optimize.cc
/*fused_activation_function=*/ rewriter.getStringAttr(add_op.getFusedActivationFunction()), /*weights_format=*/rewriter.getStringAttr(fc_op.getWeightsFormat()), /*keep_num_dims=*/rewriter.getBoolAttr(fc_op.getKeepNumDims()), /*asymmetric_quantize_inputs=*/ fc_op.getAsymmetricQuantizeInputsAttr()); rewriter.replaceOp(add_op, fc.getOutput()); return success(); } };
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 00:40:15 UTC 2024 - 102.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/uniform_quantized_stablehlo_to_tfl_pass.cc
(lhs_contracting_dims[0] == lhs_rank - 2 ? rewriter.getBoolAttr(true) : rewriter.getBoolAttr(false)); const int rhs_rank = rhs_value.getType().cast<TensorType>().getRank(); const BoolAttr adj_y = (rhs_contracting_dims[0] == rhs_rank - 1 ? rewriter.getBoolAttr(true) : rewriter.getBoolAttr(false)); Value result = op.getResult();
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/translate/import_model.cc
builder_.getBoolAttr(false)); } } auto composite_control_flow_op = [&](const std::string& name) { result.name = mlir::OperationName(get_full_op_name(name), context_); bool stateless = absl::StartsWith(node_type_name, "Stateless"); mlir::BoolAttr val = builder_.getBoolAttr(stateless);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 01 11:17:36 UTC 2024 - 183.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_ops_a_m.cc
} // Shapes are known to be compatible. ReplaceTfOpWithNewOp<Ty>(rewriter, op, op.getX(), op.getY(), rewriter.getBoolAttr(true)); return success(); } } // namespace void EqualOp::getCanonicalizationPatterns(RewritePatternSet& results, MLIRContext* context) {
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/tensorflow/transforms/shape_inference.cc
if (isa<TensorFlowDialect, BuiltinDialect>(element_type.getDialect())) return b.create<TF::CastOp>(loc, dst_type, input, /*truncate=*/b.getBoolAttr(false)); return nullptr; } // Follow the use chain of TensorList and return true iff all elements written // to TensorList have same static shape. If all elements have same shape, assign
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Jun 08 07:28:49 UTC 2024 - 134.1K bytes - Viewed (0)