Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 3 of 3 for kernel_type (0.19 sec)

  1. tensorflow/compiler/mlir/lite/tests/ops.mlir

      %0 = "tfl.lstm"(%arg0, %arg1, %arg2, %arg3, %arg4, %arg5, %arg6, %arg7, %arg8, %arg9, %arg10, %arg11, %arg12, %arg13, %arg14, %arg15, %arg16, %arg17, %arg18, %arg19, %arg20, %arg21, %arg22, %arg23) ({}) {cell_clip = 1.000000e+00 : f32, fused_activation_function = "NONE", kernel_type = #tfl<lstm_kernel_type_attr BASIC>} : (tensor<?xf32>,...
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Jun 06 19:09:08 UTC 2024
    - 189.2K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tf2xla/transforms/legalize_tf.cc

        // precise than the input types, are used for this op.
        Type kernel_type = mlir::cast<TensorType>(scale.getType()).getElementType();
        grad = rewriter.create<ConvertOp>(loc, grad, kernel_type);
        act = rewriter.create<ConvertOp>(loc, act, kernel_type);
    
        tensorflow::TensorFormat data_format;
        if (!FormatFromString(op.getDataFormat().str(), &data_format))
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jun 11 20:00:43 UTC 2024
    - 291.8K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/lite/ir/tfl_ops.td

        ConfinedAttr<
          DefaultValuedOptionalAttr<TFL_LSTMKernelTypeAttr, "mlir::TFL::LSTMKernelType::BASIC">,
          [TFL_LSTMKernelTypeEqualsTo<"mlir::TFL::LSTMKernelType::BASIC">]>:$kernel_type
      );
    
      let hasOptions = 1;
    
      let results = (outs TFL_2DTensorOf<[F32, QUI8]>:$activ_output,
                          TFL_2DTensorOf<[F32, QUI16]>:$state_output,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Jun 06 19:09:08 UTC 2024
    - 186K bytes
    - Viewed (0)
Back to top