Home
last modified time | relevance | path

Searched defs:BatchToSpaceND (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dbatch_to_space_nd.h39 inline void BatchToSpaceND(const RuntimeShape& unextended_input1_shape, in BatchToSpaceND() function
Dlegacy_reference_ops.h2044 inline void BatchToSpaceND(const T* input_data, const Dims<4>& input_dims, in BatchToSpaceND() function
/external/tensorflow/tensorflow/lite/toco/tflite/
Doperator.cc328 class BatchToSpaceND class
/external/tensorflow/tensorflow/compiler/tests/
Drandomized_tests.cc1264 TEST_F(OpTest, BatchToSpaceND) { in TEST_F() argument
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Dlegacy_optimized_ops.h4842 inline void BatchToSpaceND(const T* input_data, const Dims<4>& input_dims, in BatchToSpaceND() function
Doptimized_ops.h4986 inline void BatchToSpaceND( in BatchToSpaceND() function
/external/tensorflow/tensorflow/go/op/
Dwrappers.go23821 func BatchToSpaceND(scope *Scope, input tf.Output, block_shape tf.Output, crops tf.Output) (output … func