Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 3 of 3 for UI32 (0.03 sec)

  1. tensorflow/compiler/mlir/lite/ir/tfl_ops.td

      let arguments = (ins
         TFL_TensorOf<[I8, I16, I32, I64, UI8, UI16, UI32, UI64, F32, F64]>:$input,
         TFL_TensorOf<[I32]>:$dilations,
         TFL_0DTensorOf<[I8, I16, I32, I64, UI8, UI16, UI32, UI64, F32, F64]>:$padding_value
      );
    
      let results = (outs TFL_TensorOf<[I8, I16, I32, I64, UI8, UI16, UI32, UI64, F32, F64]>:$output);
    }
    
    def TFL_AddOp : TFL_Op<"add", [
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Jun 06 19:09:08 UTC 2024
    - 186K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/lite/tests/ops.mlir

    }
    
    // CHECK-LABEL: testMul32BitUInt
    func.func @testMul32BitUInt(tensor<? x ui32>, tensor<? x ui32>) -> tensor<? x ui32> {
    ^bb0(%arg0: tensor<? x ui32>, %arg1: tensor<? x ui32>):
      // CHECK: tfl.mul %arg0, %arg1 {fused_activation_function = "RELU6"}
      %0 = tfl.mul %arg0, %arg1 {fused_activation_function = "RELU6"} : tensor<? x ui32>
      func.return %0#0 : tensor<? x ui32>
    }
    
    // CHECK-LABEL: testMul16BitInt
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Jun 06 19:09:08 UTC 2024
    - 189.2K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/lite/ir/tfl_ops.cc

    }
    
    // Return true when the given element_type is I32.
    bool IsI32Type(Type element_type) {
      return element_type.isInteger(32) && !element_type.isUnsignedInteger();
    }
    
    // Return true when the given element_type is UI32.
    bool IsUI32Type(Type element_type) {
      return element_type.isInteger(32) && element_type.isUnsignedInteger();
    }
    
    // Return true when the given element_type is I64.
    bool IsI64Type(Type element_type) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 02 09:41:17 UTC 2024
    - 169.2K bytes
    - Viewed (0)
Back to top