| /external/tensorflow/tensorflow/lite/kernels/ |
| batch_to_space_nd.cc | 28 namespace batch_to_space_nd { namespace in namespace:tflite::ops::builtin
182 } // namespace batch_to_space_nd
186 nullptr, nullptr, batch_to_space_nd::Prepare,
187 batch_to_space_nd::Eval<batch_to_space_nd::kReference>};
193 nullptr, nullptr, batch_to_space_nd::Prepare,
194 batch_to_space_nd::Eval<batch_to_space_nd::kGenericOptimized>};
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| batchtospace_op_test.py | 145 _ = array_ops.batch_to_space_nd(
157 t = array_ops.batch_to_space_nd(input_placeholder, block_shape_placeholder,
202 _ = array_ops.batch_to_space_nd(
209 t = array_ops.batch_to_space_nd(
218 t = array_ops.batch_to_space_nd(
227 t = array_ops.batch_to_space_nd(
234 t = array_ops.batch_to_space_nd(
241 t = array_ops.batch_to_space_nd(
316 tf_y = array_ops.batch_to_space_nd(tf_x, block_shape, crops)
|
| spacetobatch_op_test.py | 210 x_tf = array_ops.batch_to_space_nd(
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| spacetobatch_op_test.py | 182 x_tf = array_ops.batch_to_space_nd(placeholder, block_shape,
|
| /external/tensorflow/tensorflow/contrib/quantize/python/ |
| fold_batch_norms.py | 481 conv = array_ops.batch_to_space_nd(
[all...] |
| /external/tensorflow/tensorflow/python/ops/ |
| array_grad.py | 663 array_ops.batch_to_space_nd(grad, op.inputs[1], op.inputs[2]), None, None
[all...] |
| array_ops.py | [all...] |
| nn_ops.py | 342 batch_to_space_nd(
377 Because `space_to_batch_nd` and `batch_to_space_nd` assume that the spatial
384 Furthermore, `space_to_batch_nd` and `batch_to_space_nd` handle this case
418 `batch_to_space_nd`.
621 result_converted = array_ops.batch_to_space_nd(
[all...] |
| /external/tensorflow/tensorflow/lite/testing/ |
| generate_examples.py | 106 r"batch_to_space_nd.*input_shape=\[8,2,2,2,1,1\]": "70594733",
[all...] |
