Home
last modified time | relevance | path

Searched refs:space_to_batch_nd (Results 1 – 25 of 27) sorted by relevance

12

/external/tensorflow/tensorflow/lite/kernels/
Dspace_to_batch_nd.cc29 namespace space_to_batch_nd { namespace
189 nullptr, nullptr, space_to_batch_nd::Prepare, in Register_SPACE_TO_BATCH_ND_REF()
190 space_to_batch_nd::Eval<space_to_batch_nd::kReference>}; in Register_SPACE_TO_BATCH_ND_REF()
196 nullptr, nullptr, space_to_batch_nd::Prepare, in Register_SPACE_TO_BATCH_ND_GENERIC_OPT()
197 space_to_batch_nd::Eval<space_to_batch_nd::kGenericOptimized>}; in Register_SPACE_TO_BATCH_ND_GENERIC_OPT()
DAndroid.bp154 "space_to_batch_nd.cc",
DBUILD650 "space_to_batch_nd.cc",
/external/tensorflow/tensorflow/core/api_def/python_api/
Dapi_def_SpaceToBatchND.pbtxt4 name: "space_to_batch_nd"
7 name: "manip.space_to_batch_nd"
/external/tensorflow/tensorflow/python/kernel_tests/array_ops/
Dspacetobatch_op_test.py204 x_tf = array_ops.space_to_batch_nd(
426 _ = array_ops.space_to_batch_nd(
437 t = array_ops.space_to_batch_nd(input_placeholder, block_shape_placeholder,
478 _ = array_ops.space_to_batch_nd(
485 t = array_ops.space_to_batch_nd(
494 t = array_ops.space_to_batch_nd(
503 t = array_ops.space_to_batch_nd(
510 t = array_ops.space_to_batch_nd(
516 t = array_ops.space_to_batch_nd(
529 array_ops.space_to_batch_nd(tf_in, block_shape, paddings)
[all …]
/external/tensorflow/tensorflow/compiler/tests/
Dspacetobatch_op_test.py158 sess.run(array_ops.space_to_batch_nd(tf_in, block_shape, paddings))
169 sess.run(array_ops.space_to_batch_nd(tf_in, block_shape, paddings))
196 x_tf = array_ops.space_to_batch_nd(placeholder, block_shape,
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.manip.pbtxt28 name: "space_to_batch_nd"
Dtensorflow.pbtxt2148 name: "space_to_batch_nd"
/external/tensorflow/tensorflow/lite/testing/op_tests/
Dspace_to_batch_nd.py91 out = tf.space_to_batch_nd(input_tensor, block_shape, paddings)
/external/tensorflow/tensorflow/security/advisory/
Dtfsa-2021-085.md9 …lob/412c7d9bb8f8a762c5b266c9e73bfa165f29aac8/tensorflow/lite/kernels/space_to_batch_nd.cc#L82-L83):
/external/tensorflow/tensorflow/lite/testing/
Dbuild_def.bzl150 "space_to_batch_nd",
261 "space_to_batch_nd",
Dgenerate_examples_lib.py161 from tensorflow.lite.testing.op_tests.space_to_batch_nd import make_space_to_batch_nd_tests
/external/tensorflow/tensorflow/lite/
Dtflite_static.bp153 "kernels/space_to_batch_nd.cc",
/external/tensorflow/tensorflow/lite/kernels/internal/
DBUILD575 "reference/space_to_batch_nd.h",
678 "reference/space_to_batch_nd.h",
/external/tensorflow/tensorflow/python/ops/parallel_for/
Darray_test.py537 return array_ops.space_to_batch_nd(x1, block_shapes, paddings)
Dpfor.py1841 output = gen_array_ops.space_to_batch_nd(inp, block_shape, paddings)
/external/tensorflow/tensorflow/python/ops/
Darray_ops.py4087 result = space_to_batch_nd(
4102 return space_to_batch_nd(input, block_shape, paddings, name)
4105 space_to_batch_v2.__doc__ = gen_array_ops.space_to_batch_nd.__doc__
Darray_grad.py961 array_ops.space_to_batch_nd(grad, op.inputs[1], op.inputs[2]), None, None
Dnn_ops.py884 input_converted = array_ops.space_to_batch_nd(
/external/tensorflow/tensorflow/lite/testing/nnapi_tflite_zip_tests/
Dtest_manifest.txt1657 space_to_batch_nd/space_to_batch_nd_block_shape=[2,2],constant_block_shape=True,constant_paddings=T…
1669 space_to_batch_nd/space_to_batch_nd_block_shape=[2,2],constant_block_shape=True,constant_paddings=T…
1681 space_to_batch_nd/space_to_batch_nd_block_shape=[2,2],constant_block_shape=True,constant_paddings=T…
1693 space_to_batch_nd/space_to_batch_nd_block_shape=[2,2],constant_block_shape=True,constant_paddings=T…
1705 space_to_batch_nd/space_to_batch_nd_block_shape=[1,3],constant_block_shape=True,constant_paddings=T…
1708 space_to_batch_nd/space_to_batch_nd_block_shape=[2,2],constant_block_shape=True,constant_paddings=T…
1729 space_to_batch_nd/space_to_batch_nd_block_shape=[1,3],constant_block_shape=True,constant_paddings=T…
1732 space_to_batch_nd/space_to_batch_nd_block_shape=[2,2],constant_block_shape=True,constant_paddings=T…
1751 space_to_batch_nd/space_to_batch_nd_block_shape=[1,3],constant_block_shape=True,constant_paddings=T…
1755 space_to_batch_nd/space_to_batch_nd_block_shape=[2,2],constant_block_shape=True,constant_paddings=T…
/external/tensorflow/tensorflow/compiler/mlir/lite/tests/
Dlegalize-tf.mlir1138 func.func @space_to_batch_nd(%arg0: tensor<1x4x4x3xf32>, %arg1: tensor<2xi32>, %arg2: tensor<2x2xi3…
1141 // CHECK-LABEL: space_to_batch_nd
1142 …// CHECK: "tfl.space_to_batch_nd"(%arg0, %arg1, %arg2) : (tensor<1x4x4x3xf32>, tensor<2xi32>, tens…
1151 // CHECK: "tfl.space_to_batch_nd"
Dops.mlir1460 // CHECK: "tfl.space_to_batch_nd"(%arg0, %arg1, %arg2)
1461 …%0 = "tfl.space_to_batch_nd"(%arg0, %arg1, %arg2) : (tensor<1x4x4x3xf32>, tensor<2xi32>, tensor<2x…
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.pbtxt1040 name: "space_to_batch_nd"
/external/tensorflow/tensorflow/compiler/mlir/tosa/tests/
Dtfl-to-tosa-pipeline.mlir1260 …%0 = "tfl.space_to_batch_nd"(%arg0, %cst, %cst_0) : (tensor<13x21x3xf32>, tensor<1xi32>, tensor<1x…
1278 …%2 = "tfl.space_to_batch_nd"(%arg0, %0, %1) : (tensor<?x241x1x80xf32>, tensor<2xi32>, tensor<2x2xi…
/external/tensorflow/tensorflow/compiler/mlir/tosa/g3doc/
Dlegalization.md4864 ### tfl.space_to_batch_nd
4871 %output = tfl.space_to_batch_nd(%input, %block_shape, %paddings)

12