- Sort Score
- Result 10 results
- Languages All
Results 1 - 4 of 4 for pack_inputs (0.17 sec)
-
tensorflow/compiler/mlir/lite/experimental/tac/transforms/device_transform_patterns.cc
LogicalResult LowerPackIntoConcatReshape::matchAndRewrite( TFL::PackOp pack_op, PatternRewriter& rewriter) const { // Pack op should have same shape type. SmallVector<Value, 5> pack_inputs(pack_op.getValues()); auto input_type = mlir::dyn_cast<RankedTensorType>(pack_inputs[0].getType()); if (!input_type) return failure(); // Figure out output shapes. SmallVector<int64_t, 4> concat_out_shape; SmallVector<int64_t, 4> pack_out_shape;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 25.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_device.cc
replicated_inputs; llvm::SmallVector<OpAsmParser::UnresolvedOperand, 8> packed_inputs; llvm::SmallVector<OpAsmParser::UnresolvedOperand, 8> region_args; llvm::SmallVector<Type, 8> region_arg_types; int32_t n = 0; Region& body = *result.addRegion(); if (ParseReplicateOpOperands(&parser, &result, &replicated_inputs, &packed_inputs, ®ion_args, ®ion_arg_types) ||
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 33.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_device_ops.td
"llvm::ArrayRef<std::pair<ValueRange, Type>>":$replicated_inputs, "ValueRange":$packed_inputs, "TypeRange":$replica_output_types)>, OpBuilder<(ins "int":$n, "std::optional<DictionaryAttr>":$devices, "llvm::ArrayRef<std::pair<ValueRange, Type>>":$replicated_inputs, "ValueRange":$packed_inputs, "TypeRange":$replica_output_types)>, ]; let hasCustomAssemblyFormat = 1; let hasVerifier = 1;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jan 23 23:53:20 UTC 2024 - 14.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/passes/tpu_cluster_formation.cc
// `num_replicas` or 1. Collect all their operands and associated type for // creating the replicate op. llvm::SmallVector<std::pair<ValueRange, Type>, 8> replicated_inputs; llvm::SmallVector<Value, 8> packed_inputs; llvm::SmallVector<mlir::TF::TPUReplicatedInputOp, 8> replicated_ops; llvm::SmallVector<mlir::TF::TPUReplicatedInputOp, 8> packed_ops; for (const auto& pos_and_input : llvm::enumerate(replicated_input_ops)) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 22:03:30 UTC 2024 - 39.3K bytes - Viewed (0)