- Sort Score
- Result 10 results
- Languages All
Results 41 - 50 of 380 for getLoc (0.13 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/host_runtime/tpu_variable_runtime_reformatting.cc
// Unformat op. auto unformat_op = builder.create<TF::TPUReshardVariablesOp>( while_op.getLoc(), llvm::ArrayRef<Type>{}, unformat_operands); WrapOpInLaunch(&builder, execute_launch.getLoc(), unformat_op, execute_launch.getDevice()); builder.create<tf_device::ReturnOp>(while_op.getLoc(), ArrayRef<Value>{}); return true; } void TPUVariableRuntimeReformattingPass::runOnOperation() {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 21.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/passes/xla_broadcast.cc
block_arg.getLoc(), DenseIntElementsAttr::get( RankedTensorType::get({1, num_replicas}, builder.getIntegerType(32)), group_assignment_val)); StringAttr reduce_op = builder.getStringAttr("Add"); StringAttr mode = builder.getStringAttr("CrossReplica"); return builder.create<XlaAllReduceOp>(block_arg.getLoc(), block_arg.getType(),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 13 18:52:07 UTC 2024 - 13.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/parallel_execute_to_islands.cc
auto yield = builder->create<tf_executor::YieldOp>( terminator->getLoc(), terminator->getOperands()); terminator->erase(); // Create new island for each region. builder->setInsertionPoint(island_op); auto execute_island = builder->create<tf_executor::IslandOp>( island_op.getLoc(), yield.getOperandTypes(), island_op.getControl().getType(), island_op.getControlInputs());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jan 19 19:47:16 UTC 2023 - 11.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/push_transpose_through_ewise.cc
auto *new_ewise_op = rewriter.create(op->getLoc(), op->getName().getIdentifier(), {tpose_arg1.getOperand(0), tpose_arg2.getOperand(0)}, new_out_type, op->getAttrs()); // Apply original tranpose to output of ewise op. auto out_tpose_op = rewriter.create<TFL::TransposeOp>( new_ewise_op->getLoc(), op->getResult(0).getType(),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 12.5K 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/tensorflow/utils/parallel_execute_util.cc
cluster_func.getLoc(), 1, output_types); cluster_func->remove(); auto& block = parallel_execute.GetRegionBlockWithIndex(0); builder->setInsertionPointToEnd(&block); builder->insert(cluster_func); cluster_func.replaceAllUsesWith(parallel_execute); builder->create<tf_device::ReturnOp>(block.getParent()->getLoc(), cluster_func.getResults());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 13 03:57:18 UTC 2023 - 2.3K 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) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/compose_uniform_quantized_type_pass.cc
CreateI8F32UniformQuantizedType( uniform_quantize_call_op.getLoc(), *rewriter.getContext(), input_scale_value, input_zero_point_value); auto input_uniform_quantize_op = rewriter.create<stablehlo::UniformQuantizeOp>( uniform_quantize_call_op.getLoc(), /*result=*/ mlir::cast<TensorType>(input_value.getType())
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 64.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/device_attribute_to_launch.cc
auto launch_op = builder.create<tf_device::LaunchOp>( op->getLoc(), builder.getStringAttr(device), /*result_types=*/op->getResultTypes()); op->replaceAllUsesWith(launch_op); launch_op.getBody().push_back(new Block); builder.setInsertionPointToEnd(&launch_op.GetBody()); auto* return_op = builder.create<tf_device::ReturnOp>(op->getLoc(), op->getResults()) .getOperation();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Oct 04 00:59:46 UTC 2022 - 2.7K bytes - Viewed (0)