- Sort Score
- Result 10 results
- Languages All
Results 1 - 6 of 6 for pack_inputs (0.35 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/lite/quantization/lite/quantize_model_test.cc
EXPECT_THAT(pack_input1->quantization->scale[0], FloatEq(pack_input2->quantization->scale[0])); EXPECT_THAT(pack_input0->quantization->zero_point[0], Eq(pack_input1->quantization->zero_point[0])); EXPECT_THAT(pack_input1->quantization->zero_point[0], Eq(pack_input2->quantization->zero_point[0])); EXPECT_THAT(pack_input1->quantization->scale[0],
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 23:15:24 UTC 2024 - 73.9K 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) -
tensorflow/compiler/mlir/lite/ir/tfl_ops.cc
for (auto input_output : llvm::zip(pack_op.getOperands(), input_unpack_op.getResults())) { Value pack_input = std::get<0>(input_output); Value unpack_output = std::get<1>(input_output); // Make sure the ordering is the same for the pack op & unpack op. if (pack_input != unpack_output) return failure(); } // Replace the pack's output to the unpack's input.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 09:41:17 UTC 2024 - 169.2K bytes - Viewed (0)