Home
last modified time | relevance | path

Searched defs:BatchToSpace (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
H A Dbatchtospace_op.cc24 void BatchToSpace(XlaOpKernelContext* ctx, const xla::XlaOp& input, in BatchToSpace() function
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dspace_to_batch_converter.cc2368 StatusOr<HloInstruction*> ConvolutionVisitor::BatchToSpace( in BatchToSpace() function in xla::__anon2cd323230111::ConvolutionVisitor
/aosp_15_r20/external/ComputeLibrary/
H A Dfilelist.json186 "BatchToSpace": { object
971 "BatchToSpace": { object
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tests/
H A Drandomized_tests.cc1883 TEST_F(OpTest, BatchToSpace) { in TEST_F() argument
/aosp_15_r20/external/tensorflow/tensorflow/go/op/
H A Dwrappers.go2916 func BatchToSpace(scope *Scope, input tf.Output, crops tf.Output, block_size int64) (output tf.Outp… func