- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for contracting_dimensions (0.26 sec)
-
tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_hlo.cc
ArrayRef<int64_t> contracting_dimensions) { const int64_t rank = type.getRank(); for (const int64_t dim : batch_dimensions) { batch_dimensions_.axes.push_back(dim); batch_dimensions_.sizes.push_back(type.getDimSize(dim)); } for (const int64_t dim : contracting_dimensions) { contracting_dimensions_.axes.push_back(dim);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 154.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/uniform_quantized_stablehlo_to_tfl_pass.cc
// * filter: per-channel qi8 // * output: per-tensor qi8 // // Conditions for the `tfl.batch_matmul` conversion: // * size(batching_dimensions) <= 3 (TFLite support restriction) // * size(contracting_dimensions) = 1 // * Input tensors are per-tensor uniform quantized (i8->f32) // tensors (full integer) with shape [..., r_x, c_x] or [..., c_x, r_x]. // * The filter tensor is a per-tensor uniform quantized (i8->f32) tensor
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/stablehlo/tests/uniform-quantized-stablehlo-to-tfl.mlir
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 14 17:10:32 UTC 2024 - 106.2K bytes - Viewed (0)