Searched refs:broadcastable_dims (Results 1 – 1 of 1) sorted by relevance
518 nvinfer1::Dims broadcastable_dims(dims); in CreateBroadcastableScalarConstant() local519 for (int i = 0; i < broadcastable_dims.nbDims; i++) { in CreateBroadcastableScalarConstant()520 broadcastable_dims.d[i] = 1; in CreateBroadcastableScalarConstant()523 broadcastable_dims); in CreateBroadcastableScalarConstant()