- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for SPACE_TO_BATCH_ND (0.35 sec)
-
tensorflow/compiler/mlir/lite/tests/legalize-tf.mlir
// CHECK: "tfl.batch_to_space_nd" } func.func @space_to_batch_nd(%arg0: tensor<1x4x4x3xf32>, %arg1: tensor<2xi32>, %arg2: tensor<2x2xi32>) -> tensor<*xf32> { %0 = "tf.SpaceToBatchND"(%arg0, %arg1, %arg2) : (tensor<1x4x4x3xf32>, tensor<2xi32>, tensor<2x2xi32>) -> tensor<*xf32> func.return %0 : tensor<*xf32> // CHECK-LABEL: space_to_batch_nd
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 05 01:54:33 UTC 2024 - 153.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/tests/ops.mlir
// ----- // CHECK-LABEL: testSpaceToBatchND func.func @testSpaceToBatchND(%arg0 : tensor<1x4x4x3xf32>, %arg1 : tensor<2xi32>, %arg2 : tensor<2x2xi32>) -> tensor<?xf32> { // CHECK: "tfl.space_to_batch_nd"(%arg0, %arg1, %arg2) %0 = "tfl.space_to_batch_nd"(%arg0, %arg1, %arg2) : (tensor<1x4x4x3xf32>, tensor<2xi32>, tensor<2x2xi32>) -> tensor<?xf32> func.return %0 : 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) -
tensorflow/compiler/mlir/lite/ir/tfl_ops.td
TFL_TensorOf<[I32]>:$indices ); let results = (outs TFL_TensorOf<[F32, I16, I32, I64, UI8, QI8, QUI8, QI16]>:$output ); } def TFL_SpaceToBatchNdOp: TFL_Op<"space_to_batch_nd", [ Pure, QuantizableResult, SameOperandsAndResultsScale, TFL_OperandHasRankRange<0, 3, 4>, PredOpTrait<"input and output must have same element type",
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 06 19:09:08 UTC 2024 - 186K bytes - Viewed (0)