Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/lite/kernels/
Dspace_to_batch_nd.cc29 namespace space_to_batch_nd { namespace
188 nullptr, nullptr, space_to_batch_nd::Prepare, in Register_SPACE_TO_BATCH_ND_REF()
189 space_to_batch_nd::Eval<space_to_batch_nd::kReference>}; in Register_SPACE_TO_BATCH_ND_REF()
195 nullptr, nullptr, space_to_batch_nd::Prepare, in Register_SPACE_TO_BATCH_ND_GENERIC_OPT()
196 space_to_batch_nd::Eval<space_to_batch_nd::kGenericOptimized>}; in Register_SPACE_TO_BATCH_ND_GENERIC_OPT()
DAndroid.bp147 "space_to_batch_nd.cc",
DBUILD622 "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/
Dspacetobatch_op_test.py206 x_tf = array_ops.space_to_batch_nd(
428 _ = array_ops.space_to_batch_nd(
439 t = array_ops.space_to_batch_nd(input_placeholder, block_shape_placeholder,
480 _ = array_ops.space_to_batch_nd(
487 t = array_ops.space_to_batch_nd(
496 t = array_ops.space_to_batch_nd(
505 t = array_ops.space_to_batch_nd(
512 t = array_ops.space_to_batch_nd(
518 t = array_ops.space_to_batch_nd(
591 tf_y = array_ops.space_to_batch_nd(tf_x, block_shape, paddings)
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.manip.pbtxt28 name: "space_to_batch_nd"
Dtensorflow.pbtxt2140 name: "space_to_batch_nd"
/external/tensorflow/tensorflow/lite/testing/op_tests/
Dspace_to_batch_nd.py95 out = tf.space_to_batch_nd(input_tensor, block_shape, paddings)
/external/tensorflow/tensorflow/compiler/tests/
Dspacetobatch_op_test.py176 x_tf = array_ops.space_to_batch_nd(placeholder, block_shape,
/external/tensorflow/tensorflow/lite/
Dtflite_static.bp130 "kernels/space_to_batch_nd.cc",
Dbuild_def.bzl389 "space_to_batch_nd",
/external/tensorflow/tensorflow/lite/testing/
Dgenerate_examples_lib.py133 from tensorflow.lite.testing.op_tests.space_to_batch_nd import make_space_to_batch_nd_tests
/external/tensorflow/tensorflow/lite/kernels/internal/
DBUILD506 "reference/space_to_batch_nd.h",
615 "reference/space_to_batch_nd.h",
/external/tensorflow/tensorflow/lite/micro/tools/make/
DMakefile353 tensorflow/lite/micro/kernels/space_to_batch_nd.cc \
440 tensorflow/lite/kernels/internal/reference/space_to_batch_nd.h \
/external/tensorflow/tensorflow/python/ops/parallel_for/
Darray_test.py512 return array_ops.space_to_batch_nd(x1, block_shapes, paddings)
Dpfor.py1823 output = gen_array_ops.space_to_batch_nd(inp, block_shape, paddings)
/external/tensorflow/tensorflow/python/ops/
Darray_ops.py3961 result = space_to_batch_nd(
3976 return space_to_batch_nd(input, block_shape, paddings, name)
3979 space_to_batch_v2.__doc__ = gen_array_ops.space_to_batch_nd.__doc__
Darray_grad.py935 array_ops.space_to_batch_nd(grad, op.inputs[1], op.inputs[2]), None, None
Dnn_ops.py724 input_converted = array_ops.space_to_batch_nd(
/external/tensorflow/tensorflow/lite/micro/kernels/
DBUILD242 "space_to_batch_nd.cc",
/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.mlir1091 func @space_to_batch_nd(%arg0: tensor<1x4x4x3xf32>, %arg1: tensor<2xi32>, %arg2: tensor<2x2xi32>) -…
1094 // CHECK-LABEL: space_to_batch_nd
1095 …// CHECK: "tfl.space_to_batch_nd"(%arg0, %arg1, %arg2) : (tensor<1x4x4x3xf32>, tensor<2xi32>, tens…
1104 // CHECK: "tfl.space_to_batch_nd"
Dops.mlir1378 // CHECK: "tfl.space_to_batch_nd"(%arg0, %arg1, %arg2)
1379 …%0 = "tfl.space_to_batch_nd"(%arg0, %arg1, %arg2) : (tensor<1x4x4x3xf32>, tensor<2xi32>, tensor<2x…
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.pbtxt1032 name: "space_to_batch_nd"
/external/tensorflow/tensorflow/compiler/mlir/tosa/tests/
Dtfl-to-tosa-pipeline.mlir696 …%0 = "tfl.space_to_batch_nd"(%arg0, %cst, %cst_0) : (tensor<13x21x3xf32>, tensor<1xi32>, tensor<1x…

12