Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 3 of 3 for add_kernel_spatial_dimensions (0.37 sec)

  1. tensorflow/compiler/mlir/quantization/stablehlo/passes/bridge/convert_tf_quant_ops_to_mhlo.cc

      dnums.set_kernel_output_feature_dimension(
          dnums_input.kernel_output_feature_dimension());
      for (auto value : dnums_input.kernel_spatial_dimensions()) {
        dnums.add_kernel_spatial_dimensions(value);
      }
      dnums.set_output_batch_dimension(dnums_input.output_batch_dimension());
      dnums.set_output_feature_dimension(dnums_input.output_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)
  2. tensorflow/compiler/mlir/quantization/tensorflow/passes/replace_cast_hacks_with_tf_xla_ops.cc

      dnums.set_output_batch_dimension(0);
      dnums.set_output_feature_dimension(num_dims - 1);
    
      for (int64_t i : llvm::seq<int64_t>(1, num_dims - 1)) {
        dnums.add_input_spatial_dimensions(i);
        dnums.add_kernel_spatial_dimensions(i - 1);
        dnums.add_output_spatial_dimensions(i);
      }
    
      Value padding, window_strides, lhs_dilation, rhs_dilation,
          feature_group_count;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 47.1K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/lite/schema/schema_generated.h

        fbb_.AddElement<int64_t>(StablehloConvolutionOptions::VT_KERNEL_OUTPUT_FEATURE_DIMENSION, kernel_output_feature_dimension, 0);
      }
      void add_kernel_spatial_dimensions(::flatbuffers::Offset<::flatbuffers::Vector<int64_t>> kernel_spatial_dimensions) {
        fbb_.AddOffset(StablehloConvolutionOptions::VT_KERNEL_SPATIAL_DIMENSIONS, kernel_spatial_dimensions);
      }
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 21 18:21:50 UTC 2024
    - 1M bytes
    - Viewed (0)
Back to top