HomeSort by relevance Sort by last modified time
    Searched refs:space_to_batch_nd (Results 1 - 7 of 7) sorted by null

  /external/tensorflow/tensorflow/lite/kernels/
space_to_batch_nd.cc 28 namespace space_to_batch_nd { namespace in namespace:tflite::ops::builtin
177 } // namespace space_to_batch_nd
181 nullptr, nullptr, space_to_batch_nd::Prepare,
182 space_to_batch_nd::Eval<space_to_batch_nd::kReference>};
188 nullptr, nullptr, space_to_batch_nd::Prepare,
189 space_to_batch_nd::Eval<space_to_batch_nd::kGenericOptimized>};
  /external/tensorflow/tensorflow/python/kernel_tests/
spacetobatch_op_test.py 100 This uses the Python compatibility wrapper that forwards to space_to_batch_nd.
206 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(
    [all...]
  /external/tensorflow/tensorflow/compiler/tests/
spacetobatch_op_test.py 176 x_tf = array_ops.space_to_batch_nd(placeholder, block_shape,
  /external/tensorflow/tensorflow/python/ops/
array_ops.py     [all...]
array_grad.py 680 array_ops.space_to_batch_nd(grad, op.inputs[1], op.inputs[2]), None, None
    [all...]
nn_ops.py 343 op(space_to_batch_nd(input, adjusted_dilation_rate, adjusted_paddings),
377 Because `space_to_batch_nd` and `batch_to_space_nd` assume that the spatial
382 amounts are 0, then space_to_batch_nd effectively leaves that dimension alone,
384 Furthermore, `space_to_batch_nd` and `batch_to_space_nd` handle this case
417 This eliminates the overhead of `k-1` calls to `space_to_batch_nd` and
616 input_converted = array_ops.space_to_batch_nd(
    [all...]
  /external/tensorflow/tensorflow/lite/testing/
generate_examples.py     [all...]

Completed in 3190 milliseconds