Searched refs:space_to_batch (Results 1 – 14 of 14) sorted by relevance
| /external/ComputeLibrary/tests/validation/reference/ |
| D | SpaceToBatch.cpp | 38 SimpleTensor<T> space_to_batch(const SimpleTensor<T> &src, const SimpleTensor<int32_t> &block_shape… in space_to_batch() function
94 template SimpleTensor<float> space_to_batch(const SimpleTensor<float> &src, const SimpleTensor<int3…
95 template SimpleTensor<half> space_to_batch(const SimpleTensor<half> &src, const SimpleTensor<int32_…
96 template SimpleTensor<uint8_t> space_to_batch(const SimpleTensor<uint8_t> &src, const SimpleTensor<…
|
| D | SpaceToBatch.h | 39 SimpleTensor<T> space_to_batch(const SimpleTensor<T> &src, const SimpleTensor<int32_t> &block_shape…
|
| /external/tensorflow/tensorflow/python/kernel_tests/array_ops/ |
| D | spacetobatch_op_test.py | 76 def space_to_batch(*args, **kwargs): member in PythonOpImpl
77 return array_ops.space_to_batch(*args, **kwargs)
87 def space_to_batch(*args, **kwargs): member in CppOpImpl
88 return gen_array_ops.space_to_batch(*args, **kwargs)
104 x_tf = self.space_to_batch(
323 y1 = self.space_to_batch(x, paddings, block_size=block_size)
345 _ = self.space_to_batch(x_np, paddings, block_size)
354 out_tf = self.space_to_batch(x_np, paddings, block_size)
364 out_tf = self.space_to_batch(x_np, paddings, block_size)
374 out_tf = self.space_to_batch(x_np, paddings, block_size)
[all …]
|
| /external/ComputeLibrary/tests/validation/fixtures/ |
| D | SpaceToBatchFixture.h | 79 FunctionType space_to_batch; in compute_target() local
80 space_to_batch.configure(&input, &block_shape, &paddings, &output); in compute_target()
110 space_to_batch.run(); in compute_target()
132 return reference::space_to_batch(input, block_shape, paddings, output_shape); in compute_reference()
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | depthwise_conv_op_test.py | 590 t3 = array_ops.space_to_batch(
668 native_t3 = array_ops.space_to_batch(
670 native_t0_transform = array_ops.space_to_batch(
|
| D | spacetobatch_op_test.py | 76 x_tf = gen_array_ops.space_to_batch(
|
| /external/tensorflow/tensorflow/python/kernel_tests/nn_ops/ |
| D | atrous_conv2d_test.py | 128 y2 = array_ops.space_to_batch(x, paddings=pad, block_size=rate)
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | array_ops.py | 4078 def space_to_batch( # pylint: disable=missing-docstring function
4096 space_to_batch.__doc__ = gen_array_ops.space_to_batch.__doc__
|
| D | array_grad.py | 953 array_ops.space_to_batch(grad, op.inputs[1], block_size=block_size), None
|
| D | nn_ops.py | 2949 value = array_ops.space_to_batch(
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.nn.pbtxt | 320 name: "space_to_batch"
|
| D | tensorflow.pbtxt | 1036 name: "space_to_batch"
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.nn.pbtxt | 392 name: "space_to_batch"
|
| D | tensorflow.pbtxt | 2144 name: "space_to_batch"
|