- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 38 for dimension_numbers (0.23 sec)
-
tensorflow/compiler/mlir/quantization/stablehlo/passes/lift_quantizable_spots_as_functions_simple.td
def LiftGather : Pat< (StableHLO_GatherOp:$res $operand, $start_indices, $dimension_numbers, $slice_sizes, $indices_are_sorted), (LiftAsTFXlaCallModule<"composite_gather_fn"> (ArgumentList $operand, $start_indices), (ResultList $res), (NamedAttributeList (NamedAttr<"dimension_numbers"> $dimension_numbers), (NamedAttr<"slice_sizes"> $slice_sizes),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Mar 28 23:21:42 UTC 2024 - 3.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/lift_quantizable_spots_as_functions_fusion.td
(NamedAttr<"rhs_dilation"> (DefaultOrNullAttr $rhs_dilation)), (NamedAttr<"window_reversal"> (DefaultOrNullAttr $window_reversal)), (NamedAttr<"dimension_numbers"> $dimension_numbers), (NamedAttr<"feature_group_count"> $feature_group_count), (NamedAttr<"batch_group_count"> $batch_group_count), (NamedAttr<"precision_config"> (DefaultOrNullAttr $precision_config)))),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 04 07:19:09 UTC 2024 - 23.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/nchw_convolution_to_nhwc.cc
// Matches input dimensions corresponding to: [b, f, 0, 1]. bool MatchInputDimensionNumbers( const ConvDimensionNumbersAttr dimension_numbers) const { return dimension_numbers.getInputBatchDimension() == 0 && dimension_numbers.getInputFeatureDimension() == 1 && dimension_numbers.getInputSpatialDimensions() == 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/quantization/tensorflow/tests/propagate_quantize_type.mlir
%1 = "tf.PartitionedCall"(%0) {config = "", config_proto = "", executor_type = "", f = @composite_dequantize_uniform} : (tensor<2x1024xi8>) -> tensor<2x1024xbf16> %2 = "tf.XlaDotV2"(%arg0, %1) {device = "", dimension_numbers = "\12\01\00\0A\01\03", precision_config = ""} : (tensor<1x2x2x2xbf16>, tensor<2x1024xbf16>) -> tensor<1x2x2x1024xbf16> %3 = "tf.Identity"(%2) : (tensor<1x2x2x1024xbf16>) -> tensor<1x2x2x1024xbf16>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 30 06:52:55 UTC 2023 - 6.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/tests/convert_tf_xla_op_to_tf_op.mlir
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 30 06:52:55 UTC 2023 - 3.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/convert_tf_xla_op_to_tf_op.td
def ConvertXlaGatherOpWithoutBatch : Pat< (TF_XlaGatherOp:$gather $operand, $start_indices, $slice_sizes, $dimension_numbers, $indices_are_sorted), (CreateSliceAndReshapeOpFromXlaGatherOpWithoutBatch $operand, $start_indices, $slice_sizes, $gather, $dimension_numbers),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sun Dec 10 05:52:02 UTC 2023 - 2.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/convert_tf_xla_op_to_tf_op.cc
OpBuilder& builder, const Location loc, Value operand, Value start_indices, Value slice_sizes, Value output, StringAttr dimension_numbers_str) { // Reads dimension numbers. xla::GatherDimensionNumbers dimension_numbers; dimension_numbers.ParseFromString(dimension_numbers_str.str()); // Construct full start_indices with given start_indices and // start_index_map. const ArrayRef<int64_t> operand_shape =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 13.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/tests/quantize_weights.mlir
// CHECK: %[[MATMUL_1:.*]] = "tf.XlaDotV2"(%arg3, %arg4) <{dimension_numbers = "\12\01\00\0A\01\03", precision_config = ""}> {device = ""} : (tensor<1x1024xbf16>, tensor<1024x1024xbf16>) -> tensor<1x1024xbf16> // CHECK: %[[IDENTITY_2:.*]] = "tf.Identity"(%arg4) {device = ""} : (tensor<1024x1024xbf16>) -> tensor<1024x1024xbf16>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 30 06:52:55 UTC 2023 - 42K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/uniform_quantized_stablehlo_to_tfl_pass.cc
return stablehlo::ConvDimensionNumbersAttr::get( context, dimension_numbers.getInputBatchDimension(), dimension_numbers.getInputFeatureDimension(), dimension_numbers.getInputSpatialDimensions(), kernel_input_feature_dim, kernel_output_feature_dim, kernel_spatial_dims, dimension_numbers.getOutputBatchDimension(), dimension_numbers.getOutputFeatureDimension(),
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/tf2xla/transforms/legalize_tf_patterns.td
$dimension_numbers, $indices_are_sorted), (MHLO_GatherOp $operand, $start_indices, (ToGatherDimNumsAttr $dimension_numbers), (CastElementsToI64Elements $slice_sizes), $indices_are_sorted), [(HasValidGatherDims $dimension_numbers)]>;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon May 06 18:46:23 UTC 2024 - 34.8K bytes - Viewed (0)