- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for set_kernel_output_feature_dimension (0.5 sec)
-
tensorflow/compiler/mlir/quantization/stablehlo/passes/bridge/convert_tf_quant_ops_to_mhlo.cc
dnums.add_input_spatial_dimensions(value); } dnums.set_kernel_input_feature_dimension( dnums_input.kernel_input_feature_dimension()); dnums.set_kernel_output_feature_dimension( dnums_input.kernel_output_feature_dimension()); for (auto value : dnums_input.kernel_spatial_dimensions()) { dnums.add_kernel_spatial_dimensions(value); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 17:58:54 UTC 2024 - 30.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/replace_cast_hacks_with_tf_xla_ops.cc
dnums.set_input_feature_dimension(num_dims - 1); // Kernel: [K, K, I, O] for Conv2D or [K, K, K, I, O] for Conv3D. dnums.set_kernel_input_feature_dimension(num_dims - 2); dnums.set_kernel_output_feature_dimension(num_dims - 1); // Output: [N, H, W, C] for Conv2D or [N, D, H, W, C] for Conv3D. dnums.set_output_batch_dimension(0); dnums.set_output_feature_dimension(num_dims - 1);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 47.1K bytes - Viewed (0)