Home
last modified time | relevance | path

Searched refs:NCHWToNCHW_VECT_C (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/lib/
Ddata_format.h33 xla::StatusOr<xla::XlaOp> NCHWToNCHW_VECT_C(xla::XlaOp input);
Ddata_format.cc86 xla::StatusOr<xla::XlaOp> NCHWToNCHW_VECT_C(xla::XlaOp input) { in NCHWToNCHW_VECT_C() function
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Ddepthtospace_op.cc172 auto output_reshaped = NCHWToNCHW_VECT_C(output); in Compile()
Dspacetodepth_op.cc176 auto output_reshaped = NCHWToNCHW_VECT_C(output); in Compile()