- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 11 for SpaceToDepth (0.21 sec)
-
tensorflow/compiler/mlir/tensorflow/tests/tpu_space_to_depth_pass.mlir
%1 = "tf.IteratorGetNext"(%arg5) {device = "/job:localhost/replica:0/task:0/device:CPU:0"} : (tensor<!tf_type.resource>) -> tensor<2x224x224x3xf32> // CHECK-DAG: %[[SPACETODEPTH0:.*]] = "tf.SpaceToDepth"([[INPUT:.*]]) <{block_size = 2 : i64, data_format = "NHWC"}> : (tensor<2x224x224x3xf32>) -> tensor<2x112x112x12xf32> %2 = "tf.AddV2"(%arg2, %arg3) {device = ""} : (tensor<i32>, tensor<i32>) -> tensor<i32>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 30 06:52:55 UTC 2023 - 37.4K bytes - Viewed (0) -
tensorflow/cc/gradients/array_grad.cc
TF_RETURN_IF_ERROR( GetNodeAttr(op.node()->attrs(), "block_size", &block_size)); grad_outputs->push_back(DepthToSpace(scope, grad_inputs[0], block_size)); return scope.status(); } REGISTER_GRADIENT_OP("SpaceToDepth", SpaceToDepthGrad); Status DepthToSpaceGrad(const Scope& scope, const Operation& op, const std::vector<Output>& grad_inputs, std::vector<Output>* grad_outputs) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Oct 10 23:33:32 UTC 2023 - 31.7K bytes - Viewed (0) -
tensorflow/cc/gradients/array_grad_test.cc
} TEST_F(ArrayGradTest, SpaceToDepthGrad) { TensorShape x_shape({1, 2, 2, 1}); auto x = Placeholder(scope_, DT_FLOAT, Placeholder::Shape(x_shape)); TensorShape y_shape({1, 1, 1, 4}); auto y = SpaceToDepth(scope_, x, /* block_size */ 2); RunTest(x, x_shape, y, y_shape); } TEST_F(ArrayGradTest, DepthToSpaceGrad) { TensorShape x_shape({1, 1, 1, 4});
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Oct 10 23:33:32 UTC 2023 - 19.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tpu_space_to_depth_pass.cc
pad_op = dyn_cast_or_null<TF::PadOp>(next_input.getDefiningOp()); cast_op = dyn_cast_or_null<TF::CastOp>(next_input.getDefiningOp()); } return std::nullopt; } // Checks if a convoluton can apply SpaceToDepth transform. // Only the first convolution in the graph whose batch size smaller than 8 // and its input feature size smaller than 8 can be transformed. std::optional<BlockArgumentInfo> GetConv2DInputArgNum(TF::Conv2DOp conv2d) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 29.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/ir/tfl_ops.td
SameOperandsAndResultsScale, PredOpTrait<"input and output must have same element type", TFL_TCresVTEtIsSameAsOp<0, 0>>, TFL_OperandHasRankAtMost<0, 4> ]> { let summary = "SpaceToDepth operator"; let description = [{ Rearranges blocks of spatial data, into depth. More specifically, this op outputs a copy of the input tensor where values from the `height`
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 06 19:09:08 UTC 2024 - 186K bytes - Viewed (0) -
tensorflow/compiler/jit/mark_for_compilation_pass.cc
"SegmentProdV2", "SegmentMinV2", "SegmentMaxV2", "SelfAdjointEigV2", "SoftmaxCrossEntropyWithLogits", "SpaceToBatch", "SpaceToBatchND", "SpaceToDepth", "SparseMatMul", "SparseToDense", "StackCloseV2", "StackPopV2", "StackPushV2", "StackV2", "StatefulPartitionedCall", "StatefulStandardNormalV2",
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 21 12:19:41 UTC 2024 - 85.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tf_passes.td
```mlir module { func @while_body { %input = "tf.IteratorGetNext"(...) {device = "/CPU:0"} -> tensor<2x224x224x3xf32> %space_to_depth = "tf.SpaceToDepth"(%input) {block_size = 2, ...}: (tensor<2x224x224x3xf32>) -> tensor<2x112x112x12xf32> %device_launch = "tf_device.cluster_func"(%space_to_depth,...) {func = @_func,...) return ... }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 21:18:05 UTC 2024 - 99.6K bytes - Viewed (0) -
RELEASE.md
* Fixes a division by zero in optimized pooling implementations in TFLite ([CVE-2021-29586](https://cve.mitre.org/cgi-bin/cvename.cgi?name=CVE-2021-29586)) * Fixes a division by zero in TFLite's implementation of `SpaceToDepth` ([CVE-2021-29587](https://cve.mitre.org/cgi-bin/cvename.cgi?name=CVE-2021-29587)) * Fixes a division by zero in TFLite's implementation of `GatherNd`
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 23:24:08 UTC 2024 - 730.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/g3doc/_includes/tf_passes.md
``` The program will be transformed into: ```mlir module { func @while_body { %input = "tf.IteratorGetNext"(...) {device = "/CPU:0"} -> tensor<2x224x224x3xf32> %space_to_depth = "tf.SpaceToDepth"(%input) {block_size = 2, ...}: (tensor<2x224x224x3xf32>) -> tensor<2x112x112x12xf32> %device_launch = "tf_device.cluster_func"(%space_to_depth,...) {func = @_func,...) return ... }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Aug 02 02:26:39 UTC 2023 - 96.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/tests/legalize-tf.mlir
// CHECK-LABEL: argmax64 // CHECK: "tfl.arg_max"(%arg0, %arg1) : (tensor<3xi32>, tensor<i32>) -> tensor<i64> } func.func @space_to_depth(%arg0: tensor<1x2x2x1xf32>) -> tensor<?xf32> { %0 = "tf.SpaceToDepth"(%arg0) {block_size = 2: i64, data_format = "NHWC"}: (tensor<1x2x2x1xf32>) -> tensor<?xf32> func.return %0 : tensor<?xf32> // CHECK-LABEL: space_to_depth // CHECK: %[[ARG:.*]]: tensor<1x2x2x1xf32>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 05 01:54:33 UTC 2024 - 153.4K bytes - Viewed (0)