- Sort Score
- Result 10 results
- Languages All
Results 61 - 70 of 330 for setLoc (0.14 sec)
-
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/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/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/replicate_to_island.cc
builder.create<tf_executor::YieldOp>(terminator.getLoc(), terminator.getOperands()); terminator.erase(); builder.setInsertionPoint(island_op); IRMapping mapping; for (int i : llvm::seq<int>(0, num_replicas)) { // Create new island for replica. auto replica = builder.create<tf_executor::IslandOp>( island_op.getLoc(), output_types, control_type, replica_inputs);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Jul 24 21:01:40 UTC 2023 - 16.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/breakup-islands.cc
tf_executor::YieldOp yield = island.GetYield(); if (yield.getNumOperands() == 0) { builder.create<TF::NoOp>(island.getLoc(), TypeRange{}, ValueRange{}); } else if (yield.getNumOperands() == 1) { Value operand = yield.getOperand(0); auto identity = builder.create<TF::IdentityOp>(island.getLoc(), operand.getType(), operand); yield.setOperand(0, identity.getOutput());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Aug 11 20:52:36 UTC 2023 - 16.7K 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) -
tensorflow/compiler/mlir/tensorflow/transforms/tf_data_optimization.cc
auto num_parallel_calls_op = rewriter.create<CastOp>( op.getLoc(), UnrankedTensorType::get(rewriter.getIntegerType(64)), batchInputOp.getNumParallelCalls(), rewriter.getBoolAttr(false)); if (op.getMetadata() != batchInputOp.getMetadata()) { return failure(); } auto fused_op = rewriter.create<MapAndBatchDatasetOp>( op.getLoc(), op.getType(), batchInputOp.getInputDataset(),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Nov 03 12:35:38 UTC 2022 - 2.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/rewrite_tpu_embedding_ops.cc
SendTPUEmbeddingGradientsOp send_op; if (failed(GetOp(region, &send_op))) return failure(); // No TPU embedding ops. if (!recv_op && !send_op) return success(); Location loc = recv_op ? recv_op.getLoc() : send_op.getLoc(); StringRef config = recv_op ? recv_op.getConfig() : send_op.getConfig(); // Create XlaRecvTPUEmbeddingDeduplicationData op. OpBuilder builder(region); auto output_ty =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 15 22:55:42 UTC 2024 - 4.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/cluster_outlining.cc
void runOnOperation() override; }; void ReplaceClusterReturnWithReturn(tf_device::ReturnOp cluster_return_op, OpBuilder* builder) { builder->create<func::ReturnOp>(cluster_return_op.getLoc(), cluster_return_op.getOperands()); cluster_return_op.erase(); } // Builds a function that outlines region attached to cluster_op or launch_op,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 21:25:12 UTC 2024 - 7.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/lower_globals_to_ml_program.cc
} else { initial_value = mlir::Attribute(); } opToName[globalTensor] = name; auto variableOp = globalBuilder.create<ml_program::GlobalOp>( globalTensor.getLoc(), name, globalTensor.getType(), globalTensor.getIsMutable(), initial_value, /*visibility=*/globalBuilder.getStringAttr("private")); variableOp.setPrivate(); } SymbolTable syms(module);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 8.6K bytes - Viewed (0)