Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 3 of 3 for rng_algorithm (0.24 sec)

  1. tensorflow/compiler/mlir/lite/flatbuffer_operator.cc

            algorithm = mlir::vhlo::RngAlgorithmV1::DEFAULT;
        }
        auto attr =
            mlir::vhlo::RngAlgorithmV1Attr::get(builder.getContext(), algorithm);
        attributes.emplace_back(builder.getNamedAttr("rng_algorithm", attr));
        return;
      }
    }
    
    // Pull in FlatBuffer writers for TFLite generated using TableGen
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 21 18:21:50 UTC 2024
    - 38K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/lite/flatbuffer_export.cc

      tflite::RngAlgorithm algorithm = tflite::RngAlgorithm_DEFAULT;
      switch (rng_op.getRngAlgorithm()) {
        case mlir::stablehlo::RngAlgorithm::THREE_FRY:
          algorithm = tflite::RngAlgorithm_THREEFRY;
          break;
        case mlir::stablehlo::RngAlgorithm::PHILOX:
          algorithm = tflite::RngAlgorithm_PHILOX;
          break;
        case mlir::stablehlo::RngAlgorithm::DEFAULT:
          break;
      }
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Jun 12 21:41:49 UTC 2024
    - 164.5K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/lite/schema/schema_generated.h

      return EnumNamesStablehloComparisonType()[index];
    }
    
    enum RngAlgorithm : int8_t {
      RngAlgorithm_DEFAULT = 0,
      RngAlgorithm_PHILOX = 1,
      RngAlgorithm_THREEFRY = 2,
      RngAlgorithm_MIN = RngAlgorithm_DEFAULT,
      RngAlgorithm_MAX = RngAlgorithm_THREEFRY
    };
    
    inline const RngAlgorithm (&EnumValuesRngAlgorithm())[3] {
      static const RngAlgorithm values[] = {
        RngAlgorithm_DEFAULT,
        RngAlgorithm_PHILOX,
    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