- Sort Score
- Result 10 results
- Languages All
Results 1 - 8 of 8 for GetOperands (0.14 sec)
-
tensorflow/compiler/mlir/lite/transforms/lower_static_tensor_list.cc
TF::TensorListSetItemOp op, OpAdaptor adaptor, ConversionPatternRewriter &rewriter) const { Location loc = op.getLoc(); Value input = adaptor.getOperands()[0]; Value index = adaptor.getOperands()[1]; Value item = adaptor.getOperands()[2]; IntegerType shape_dtype = rewriter.getIntegerType(32); auto item_rank = rewriter.create<TF::RankOp>(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 20:00:43 UTC 2024 - 70.7K bytes - Viewed (0) -
tensorflow/compiler/jit/deadness_analysis.cc
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 12 06:33:33 UTC 2024 - 60.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/prepare_tf.cc
Operation::operand_range mean(fused_batch_norm->getOperands()); ::mlir::FloatAttr exponential_avg_factor; ::mlir::TF::FusedBatchNormV3Op root; Operation::operand_range offset(fused_batch_norm->getOperands()); Operation::operand_range x(fused_batch_norm->getOperands()); Operation::operand_range scale(fused_batch_norm->getOperands()); Operation::operand_range variance(fused_batch_norm->getOperands());
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/tensorflow/transforms/resource_op_lifting.cc
Block& front = region->front(); auto old_return = front.getTerminator(); assert(old_return->getNumOperands() == op_->getNumResults()); auto new_return_operands = llvm::to_vector<4>(old_return->getOperands()); new_return_operands.resize(num_new_results_); // initialize return values for written resources to be the hoisted reads. for (Value resource : written_resources_) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 55.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/passes/extract_outside_compilation.cc
} } return false; } bool HasDynamicExternalValues(Operation* op) { return op ->walk([](Operation* walked_op) { for (Value v : walked_op->getOperands()) { if (mlir::TF::CanBeRefined(v.getType())) { return WalkResult::interrupt(); } } return WalkResult::advance(); }) .wasInterrupted(); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 21:25:12 UTC 2024 - 68.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/sparsecore/embedding_pipelining.cc
llvm::SetVector<Operation*> new_ops; while (!ops_to_process.empty()) { for (Operation* op : ops_to_process) { if (predecessors) { for (Value operand : op->getOperands()) { // Stop at the block boundary. if (mlir::isa<BlockArgument>(operand)) continue; Operation* predecessor = operand.getDefiningOp(); if (!operations->contains(predecessor) &&
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 92.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/flatbuffer_import.cc
mlir::Operation& return_op = region.back().back(); mlir::Location loc = return_op.getLoc(); op_builder.setInsertionPointToEnd(®ion.back()); op_builder.create<mlir::stablehlo::ReturnOp>(loc, return_op.getOperands()); return_op.erase(); } void InlineVhloOpRegion(mlir::Region& region, mlir::func::FuncOp func) { OpBuilder op_builder{region}; mlir::IRMapping mapper;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 21 18:21:50 UTC 2024 - 66.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/compose_uniform_quantized_type_pass.cc
output_uniform_quantized_tensor_type}; auto new_conv_op_with_output_type = rewriter.create<stablehlo::ConvolutionOp>( op.getLoc(), new_conv_output_types, op.getOperands(), op->getAttrs()); rewriter.replaceAllUsesWith(op.getResult(), new_conv_op_with_output_type.getResult()); auto new_output_dequant_op =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 64.6K bytes - Viewed (0)