Searched defs:broadcast_num_dims (Results 1 – 1 of 1) sorted by relevance
317 nvinfer1::Dims* output_dims) -> Status { in GetTrtBroadcastShape()345 const int broadcast_num_dims = in GetTrtBroadcastShape() local