Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/tf2xla/lib/
Ddata_format.h28 xla::StatusOr<xla::XlaOp> NCHW_VECT_CToNCHW(xla::XlaOp input);
Ddata_format.cc79 xla::StatusOr<xla::XlaOp> NCHW_VECT_CToNCHW(xla::XlaOp input) { in NCHW_VECT_CToNCHW() function
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Ddepthtospace_op.cc48 auto input_reshaped = NCHW_VECT_CToNCHW(input); in Compile()
Dspacetodepth_op.cc48 auto input_reshaped = NCHW_VECT_CToNCHW(input); in Compile()