- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for set_input_batch_dimension (0.33 sec)
-
tensorflow/compiler/mlir/quantization/stablehlo/passes/bridge/convert_tf_quant_ops_to_mhlo.cc
xla::ConvolutionDimensionNumbers ConvertConvolutionDimensionNumbers( const ::tensorflow::UniformQuantizedConvolutionDimensionNumbersAttr &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()) {
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
num_dims)); 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);
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
enum FlatBuffersVTableOffset FLATBUFFERS_VTABLE_UNDERLYING_TYPE { 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,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 21 18:21:50 UTC 2024 - 1M bytes - Viewed (0)