- Sort Score
- Result 10 results
- Languages All
Results 31 - 40 of 86 for output_shapes (0.15 sec)
-
tensorflow/compiler/mlir/lite/utils/lstm_utils_test.cc
SmallVector<int64_t, 2> output_shape{1, mlir::ShapedType::kDynamic}; EXPECT_EQ(mlir::cast<RankedTensorType>(output_types[0]).getShape().size(), output_shape.size()); for (int i = 0; i < output_shape.size(); i++) { EXPECT_EQ(mlir::cast<RankedTensorType>(output_types[0]).getDimSize(i), output_shape[i]); } }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 10K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/legalize_tfg.mlir
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 30 06:52:55 UTC 2023 - 2.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/tests/tf_to_corert/basic.mlir
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 08 00:18:59 UTC 2024 - 3.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/composite_utils.cc
std::array<int64_t, 4> output_shape; // NHWC <- NCHW output_shape[0] = composite_result_shape[0]; output_shape[1] = composite_result_shape[2]; output_shape[2] = composite_result_shape[3]; output_shape[3] = composite_result_shape[1]; auto input_type = mlir::cast<ShapedType>(old_op->getOperand(0).getType()); return RankedTensorType::get(output_shape, input_type.getElementType()); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 29 18:33:05 UTC 2024 - 3.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/quantize_patterns.td
// Transpose conv supports hybrid computation with quantized weights. def FoldQuantWeightsIntoTposeConv : Pat< (TFL_TransposeConvOp $output_shape, (TFL_DequantizeOp $quant_weights), $quant_input, $bias, $padding, $stride_h, $stride_w, $faf), (TFL_TransposeConvOp $output_shape, $quant_weights, $quant_input, $bias, $padding, $stride_h, $stride_w, $faf),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 28 23:10:13 UTC 2024 - 2.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/convert_tensor_test.cc
PartialTensorShape output_shape = ConvertTypeToTensorShape(mlir::UnrankedTensorType::get(b.getF32Type())); EXPECT_TRUE(output_shape.IsIdenticalTo(PartialTensorShape())); } TEST(ConvertTypeToTensorTypeTest, NonFullyDefinedRankedTensorType) { mlir::MLIRContext context; RegisterDialects(context); mlir::Builder b(&context); PartialTensorShape output_shape = ConvertTypeToTensorShape(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 10.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/utils/lstm_utils.cc
return failure(); // Build the lstm op. SmallVector<int64_t, 3> output_shape; if (time_majored) { output_shape = {time, batch, n_output}; } else { output_shape = {batch, time, n_output}; } auto result_type = mlir::RankedTensorType::get( output_shape, mlir::cast<RankedTensorType>(final_inputs.getType()).getElementType());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 36.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/post_quantize.cc
SmallVector<int64_t, 4> output_shape; for (int i = 0; i < num_dimensions; ++i) { perm.push_back(perm_tensor.getValues<IntegerAttr>()[i].getInt()); output_shape.push_back(input_shape[perm[i]]); // Check that the derived output shape matches the static shape. assert(!output_type.hasStaticShape() || output_type.getShape()[i] == output_shape[i]); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 17.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/graphdef2mlir/graph-library.pbtxt
# RUN: tf-mlir-translate -graphdef-to-mlir -tf-enable-shape-inference-on-import=false %s -o - | FileCheck %s node { name: "unnamed" op: "foo" attr { key: "_output_shapes" value { list { shape { } } } } attr { key: "_disable_call_shape_inference" value { b: true } } experimental_debug_info { } } node { name: "unnamed1"
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 30 06:52:55 UTC 2023 - 1.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/utils/utils.h
inline DenseElementsAttr GetShape(Value output_val, bool truncate = false) { auto output_shape = output_val.getType().dyn_cast<ShapedType>().getShape(); SmallVector<int32_t> shape; shape.reserve(output_shape.size()); bool needs_truncation = true; for (size_t dim_idx = 0; dim_idx < output_shape.size(); ++dim_idx) { int64_t dim = output_shape[dim_idx]; if (truncate && needs_truncation && dim == 1) { continue;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 00:40:15 UTC 2024 - 11.6K bytes - Viewed (0)