- Sort Score
- Result 10 results
- Languages All
Results 1 - 6 of 6 for fully_connected (0.31 sec)
-
tensorflow/compiler/mlir/lite/stablehlo/tests/uniform-quantized-stablehlo-to-tfl.mlir
// CHECK: return %[[FULLY_CONNECTED]] // ----- // Tests that when the weight tensor for `stablehlo.dot_general` has a // `stablehlo.constant` -> `stablehlo.transpose` pattern, the // `stablehlo.constant` is directly transformed to `tfl.pseudo_qconst`, which // becomes the rhs of `tfl.fully_connected`. This is because // `tfl.fully_connected` accepts a [o, i] format for rhs, which
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 14 17:10:32 UTC 2024 - 106.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/uniform_quantized_stablehlo_to_tfl_pass.cc
// filter value from [i, o] -> [o, i]. This is because we assume `[i, o]` // format for `stablehlo.dot_general` (i.e. contracting dimension == 1) // whereas `tfl.fully_connected` accepts an OI format. TFL::QConstOp CreateTransposedTflConstOpForFilter( stablehlo::ConstantOp filter_constant_op, PatternRewriter& rewriter, bool is_per_channel) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 22 09:00:19 UTC 2024 - 99.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/optimize.cc
// Remove Reshape before FullyConnected when `keep_num_dims=false` and Reshape // does not alter the last dimension as FullyConnected will collapse all other // dimensions into a single dimension. For example, // // %shape = arith.constant dense<[1, 128, 64]> : tensor<3xi32> // %reshape = tfl.reshape(%input, %shape) // %input: tensor<128x64xf32> // %fc = tfl.fully_connected(%reshape, %filter, %bias)
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 00:40:15 UTC 2024 - 102.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/tests/ops.mlir
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 06 19:09:08 UTC 2024 - 189.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/tests/legalize-tf.mlir
// CHECK: %[[ARG:.*]] = "tfl.transpose"(%arg1, %[[CST]]) : (tensor<37x40xf32>, tensor<2xi32>) -> tensor<40x37xf32> // CHECK: %[[CST_0:.*]] = "tfl.no_value"() <{value}> : () -> none // CHECK: "tfl.fully_connected"(%arg0, %[[ARG]], %[[CST_0]]) <{fused_activation_function = "NONE", keep_num_dims = false, weights_format = "DEFAULT"}> : (tensor<40x37xf32>, tensor<40x37xf32>, none) -> tensor<40x40xf32> }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 05 01:54:33 UTC 2024 - 153.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/ir/tfl_ops.td
std::vector<int> GetQuantizableOperandIndices() { return {1}; } }]; } // TODO(jpienaar): Update post discussion on semantics of FC OP. def TFL_FullyConnectedOp : TFL_Op<"fully_connected", [ Pure, AccumulatorUniformScale<2, 0, 1>, AffineQuantizedOpInterface, AffineOpCoefficient<0, 1>, TFL_SparseOp, DeclareOpInterfaceMethods<TFL_ArithmeticCount>, QuantizableResult,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 06 19:09:08 UTC 2024 - 186K bytes - Viewed (0)