- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 10 for getOperands (0.23 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
auto new_operands = llvm::to_vector<8>(caller->getOperands()); llvm::SmallVector<int64_t, 8> changed_indices; // Find the operands to change, and create the loads. for (auto& entry : arg_data_type_and_updated_output_index) { int64_t index = entry.getFirst(); Type new_type = entry.getSecond().first; int64_t updated_index = entry.getSecond().second; auto operand = caller->getOperand(index); builder.setInsertionPoint(caller);
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/tensorflow/transforms/sparsecore/embedding_pipelining.cc
func::ReturnOp return_op = *callee.getOps<func::ReturnOp>().begin(); auto val = return_op.getOperand(i); auto block_arg_number = ComesFromBlockArgNumber(val); if (block_arg_number >= 0) { replaceAllUsesInRegionWith(call_op->getResult(i), call_op->getOperand(block_arg_number), func.getRegion()); } } }
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/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/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
return failure(); } return UniformQuantizeFunctionCallPattern(call_op); } Value GetInputValue() { return call_op_.getOperand(0); } Value GetInverseScalesValue() { return call_op_.getOperand(1); } Value GetZeroPointsValue() { return call_op_.getOperand(2); } stablehlo::ConstantOp GetZeroPointsConstantOp() { return cast<stablehlo::ConstantOp>(GetZeroPointsValue().getDefiningOp()); }
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/quantization/tensorflow/passes/quantize_composite_functions.cc
int weight_operand_idx = 1; Operation* weight_op = op.getOperand(weight_operand_idx).getDefiningOp(); auto weight_type = mlir::dyn_cast<ShapedType>(weight_op->getResult(0).getType()); auto input_type = mlir::dyn_cast<ShapedType>(op.getOperand(0).getType()); llvm::ArrayRef<int64_t> weight_shape = weight_type.getShape();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 54.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/optimize_patterns.td
class AllUsersAreAddOpsAndCanFuseWith<string op_to_fuse_with> : Constraint<CPred< "llvm::all_of($0.getUsers(), [&](Operation *user) {" " if(!llvm::isa<TFL::AddOp>(user)) return false;" " if(!user->getOperand(0).getDefiningOp<"#op_to_fuse_with#">()) return false;" " return true;" "})">, "all users are AddOp and can fuse with "#op_to_fuse_with#"">; // TODO(b/294385379): This pattern only appears when we convert
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 16 20:31:41 UTC 2024 - 66.4K bytes - Viewed (0)