Searched defs:device_shape_map (Results 1 – 1 of 1) sorted by relevance
974 …const std::map<std::string, DeviceShapeTransfer> device_shape_map{{kOpFormat_NCHW, NchwDeviceShape… in TransShapeToDevice() local1020 const std::map<std::string, DeviceShapeTransfer> device_shape_map{ in TransShapeToDevice() local