Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 1 of 1 for GpuBasicSupportedOpNoCost (0.18 sec)

  1. tensorflow/compiler/mlir/lite/experimental/tac/hardwares/gpu_hardware.cc

    // tfl.relu / tfl.relu6 / tfl.rsqrt / tfl.sin / tfl.slice / tfl.softmax /
    // tfl.space_to_depth / tfl.sqrt / tfl.square / tfl.squared_difference /
    // tfl.strided_slice / tfl.tanh / tfl.transpose / tfl.transpose_conv
    class GpuBasicSupportedOpNoCost : public TargetHardwareOperation {
      double GetOpCost(mlir::Operation* op) const override { return 0; }
    
      bool IsOpSupported(mlir::Operation* op) const override {
        InferenceType inference_type = GetInferenceType(op);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jun 06 03:08:33 UTC 2023
    - 7.8K bytes
    - Viewed (0)
Back to top