- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for set_input_feature_dimension (0.77 sec)
-
tensorflow/compiler/mlir/quantization/stablehlo/passes/bridge/convert_tf_quant_ops_to_mhlo.cc
&dnums_input) { xla::ConvolutionDimensionNumbers dnums; dnums.set_input_batch_dimension(dnums_input.input_batch_dimension()); dnums.set_input_feature_dimension(dnums_input.input_feature_dimension()); for (auto value : dnums_input.input_spatial_dimensions()) { dnums.add_input_spatial_dimensions(value); } dnums.set_kernel_input_feature_dimension(
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
return {}; } xla::ConvolutionDimensionNumbers dnums; // Input: [N, H, W, C] for Conv2D or [N, D, H, W, C] for Conv3D. dnums.set_input_batch_dimension(0); 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);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 47.1K bytes - Viewed (0)