- Sort Score
- Result 10 results
- Languages All
Results 11 - 20 of 106 for setLoc (0.23 sec)
-
tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_hlo_conversions/reduce.h
auto tf_reduce_op = rewriter.create<BooleanReduce>( reduce_op.getLoc(), reduce_op->getResult(0).getType(), operand, reduction_indices, /*keep_dim=*/rewriter.getBoolAttr(false)); auto tf_argreduce_op = rewriter.create<ArgReduce>( reduce_op.getLoc(), reduce_op->getResult(1).getType(), operand, reduction_indices);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 5.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/lower_variable_ops_to_ml_program.cc
OpBuilder builder(op); Operation* load = builder.create<mlir::ml_program::GlobalLoadOp>( op.getLoc(), globalOp.getType(), SymbolRefAttr::get(op->getContext(), globalOp.getSymName())); if (globalOp.getType() != op.getValue().getType()) { load = builder.create<TF::CastOp>(op.getLoc(), op.getValue().getType(), load->getResult(0)); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Mar 06 23:53:00 UTC 2024 - 7.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/initialize_variables_in_session_init.cc
builder.insert(var_handle_op_in_init); auto const_op = builder.create<mlir::arith::ConstantOp>( session_init_func.getLoc(), tensor_attr.getType(), tensor_attr); builder.create<TF::AssignVariableOp>( session_init_func.getLoc(), llvm::ArrayRef<mlir::Type>{}, llvm::ArrayRef<mlir::Value>{var_handle_op_in_init->getResult(0), const_op.getResult()}); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 23 09:05:47 UTC 2024 - 7K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/optimize_batch_matmul.cc
Type output_type = bmm_op.getResult().getType(); auto no_input = rewriter.create<TFL::NoValueOp>( bmm_op->getLoc(), rewriter.getNoneType(), rewriter.getUnitAttr()); auto fc_op = rewriter.create<TFL::FullyConnectedOp>( bmm_op->getLoc(), ArrayRef<Type>{output_type}, /*input=*/output_lhs, /*filter=*/output_rhs, /*bias=*/no_input,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 9.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/executor_tpuv1_outline_tpu_island.cc
replacer.create<mlir::func::ReturnOp>(yield_op.getLoc(), yield_op.getOperands()); yield_op.erase(); } // Remap the captured operands in the (former) island block with newly // created entry block arguments in the function body. { Block &entry_block = outlined_func.getBody().front(); auto loc = outlined_func.getLoc(); for (Value operand : operands) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 8.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/nchw_convolution_to_nhwc.cc
mlir::cast<TensorType>(input.getType()), kNchwToNhwcPermutation); auto input_transpose_op = rewriter.create<mlir::stablehlo::TransposeOp>( op.getLoc(), /*resultType0=*/new_input_tensor_type, /*operand=*/input, rewriter.getDenseI64ArrayAttr(kNchwToNhwcPermutation)); // Transpose the filter tensor: [o, i, 0, 1] => [0, 1, i, o] Value filter = op->getOperand(1);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 8.2K bytes - Viewed (0) -
maven-plugin-api/src/main/java/org/apache/maven/plugin/Mojo.java
* It features an <code>execute()</code> method, which triggers the Mojo's build-process behavior, and can throw * a MojoExecutionException or MojoFailureException if error conditions occur.<br> * Also included is the <code>setLog(...)</code> method, which simply allows Maven to inject a logging mechanism which * will allow the Mojo to communicate to the outside world through standard Maven channels. * */ public interface Mojo {
Registered: Wed Jun 12 09:55:16 UTC 2024 - Last Modified: Wed Sep 06 08:39:32 UTC 2023 - 2.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/fuse_convolution_pass.cc
} Value broadcast_multiplier = rewriter.create<mhlo::BroadcastInDimOp>( mul_op.getLoc(), filter.getType(), multiplier, broadcast_dims); Value new_filter = rewriter.create<mhlo::MulOp>( mul_op.getLoc(), filter.getType(), filter, broadcast_multiplier); Value new_conv = rewriter.create<mhlo::ConvolutionOp>( mul_op.getLoc(), conv_op.getType(), conv_op.getLhs(), new_filter,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 22 22:21:19 UTC 2024 - 8.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/utils/nms_utils.cc
OpBuilder builder(func_.getBody()); auto op = builder.create<mlir::TFL::NonMaxSuppressionV4Op>( func_.getLoc(), output_type0, output_type1, boxes, scores, max_output_size, iou_threshold, score_threshold); builder.create<mlir::func::ReturnOp>(func_.getLoc(), op.getResults()); } LogicalResult ConvertNMSPaddedFunc::VerifySignature() { // Verify high-level function signature.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 8.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/unfold_large_splat_constant.cc
const_op->getLoc(), DenseElementsAttr::get( tensorflow::GetTypeFromTFTensorShape( {}, splat_elements_attr.getType().getElementType()), splat_elements_attr.getSplatValue<Attribute>())); TFL::FillOp fill = op_builder->create<TFL::FillOp>( const_op->getLoc(), splat_elements_attr.getType(), fill_shape, fill_value);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 4.5K bytes - Viewed (0)