- Sort Score
- Result 10 results
- Languages All
Results 1 - 4 of 4 for getOutputBatchDimension (0.48 sec)
-
tensorflow/compiler/mlir/quantization/stablehlo/passes/nchw_convolution_to_nhwc.cc
} // Matches output dimensions corresponding to: [b, f, 0, 1]. bool MatchOutputDimensionNumbers( const ConvDimensionNumbersAttr dimension_numbers) const { return dimension_numbers.getOutputBatchDimension() == 0 && dimension_numbers.getOutputFeatureDimension() == 1 && dimension_numbers.getOutputSpatialDimensions() == ArrayRef<int64_t>{2, 3}; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 8.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_hlo.cc
if (dnums.getOutputBatchDimension() != 0 || dnums.getOutputFeatureDimension() != num_spatial_dims + 1) { std::vector<int64_t> transpose_order; transpose_order.resize(num_spatial_dims + 2); // the output always in b, 0, 1, f format transpose_order[dnums.getOutputBatchDimension()] = 0;
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
dimension_numbers.getInputFeatureDimension(), dimension_numbers.getInputSpatialDimensions(), kernel_input_feature_dim, kernel_output_feature_dim, kernel_spatial_dims, dimension_numbers.getOutputBatchDimension(), dimension_numbers.getOutputFeatureDimension(), dimension_numbers.getOutputSpatialDimensions()); } }; void UniformQuantizedStableHloToTflPass::runOnOperation() {
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/flatbuffer_export.cc
.getValue() .getSExtValue(); auto output_batch_dimension = mlir::cast<mlir::vhlo::IntegerV1Attr>( vhlo_op.getOutputBatchDimension()) .getValue() .getSExtValue(); auto output_feature_dimension = mlir::cast<mlir::vhlo::IntegerV1Attr>(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 21:41:49 UTC 2024 - 164.5K bytes - Viewed (0)