Searched defs:device_shape_vector (Results 1 – 1 of 1) sorted by relevance
37 void TransposeDefaultShape(const ShapeVector *host_shape_vector, ShapeVector *device_shape_vector) { in TransposeDefaultShape()43 void TransposeNCHWShape(const ShapeVector *host_shape_vector, ShapeVector *device_shape_vector) { in TransposeNCHWShape()52 void TransposeNHWCShape(const ShapeVector *host_shape_vector, ShapeVector *device_shape_vector) { in TransposeNHWCShape()