- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for set_input_feature_dimension (0.4 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) -
tensorflow/compiler/mlir/lite/schema/schema_generated.h
VT_WINDOW_STRIDES = 4, VT_PADDING = 6, VT_LHS_DILATION = 8, VT_RHS_DILATION = 10, VT_WINDOW_REVERSAL = 12, VT_INPUT_BATCH_DIMENSION = 14, VT_INPUT_FEATURE_DIMENSION = 16, VT_INPUT_SPATIAL_DIMENSIONS = 18, VT_KERNEL_INPUT_FEATURE_DIMENSION = 20, VT_KERNEL_OUTPUT_FEATURE_DIMENSION = 22, VT_KERNEL_SPATIAL_DIMENSIONS = 24,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 21 18:21:50 UTC 2024 - 1M bytes - Viewed (0)