- Sort Score
- Result 10 results
- Languages All
Results 1 - 4 of 4 for SpaceToBatchND (0.48 sec)
-
tensorflow/compiler/mlir/tensorflow/tests/tf-ops.mlir
// tf.SpaceToBatchND //===--------------------------------------------------------------------===// // Test valid tf.SpaceToBatchND // CHECK-LABEL: func @testSpaceToBatchND func.func @testSpaceToBatchND(%input: tensor<3x5x7x10xf32>, %block_shape: tensor<2xi64>, %paddings: tensor<2x2xi64>) -> tensor<?x?x?x10xf32> {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 23 14:40:35 UTC 2023 - 236.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/tests/legalize-tf.mlir
// CHECK: "tfl.cast" // 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/tensorflow/tests/shape_inference.mlir
%1 = "tf.Const"() {value = dense<2> : tensor<2x2xi32>} : () -> tensor<2x2xi32> %2 = "tf.Identity"(%1) {device = ""} : (tensor<2x2xi32>) -> tensor<2x2xi32> // CHECK: SpaceToBatchND{{.*}}-> tensor<4x98x130x128xf32> %3 = "tf.SpaceToBatchND"(%arg0, %0, %2) {device = ""} : (tensor<1x192x256x128xf32>, tensor<2xi32>, tensor<2x2xi32>) -> tensor<4x?x?x128xf32> func.return } // CHECK-LABEL: check_subtyperefinement
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jan 23 17:24:10 UTC 2024 - 167.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/ir/tfl_ops.td
SameOperandsAndResultsScale, TFL_OperandHasRankRange<0, 3, 4>, PredOpTrait<"input and output must have same element type", TFL_TCresVTEtIsSameAsOp<0, 0>> ]> { let summary = "SpaceToBatchNd operator"; let description = [{ This operation reshapes space dimensions into the "batch" dimension 0 }]; let arguments = (ins
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 06 19:09:08 UTC 2024 - 186K bytes - Viewed (0)