Searched refs:broadcastable_dims (Results 1 – 1 of 1) sorted by relevance
475 nvinfer1::Dims broadcastable_dims(dims); in CreateBroadcastableScalarConstant() local476 for (int i = 0; i < broadcastable_dims.nbDims; i++) { in CreateBroadcastableScalarConstant()477 broadcastable_dims.d[i] = 1; in CreateBroadcastableScalarConstant()480 params->weight_store->GetTempWeights(dtype, broadcastable_dims); in CreateBroadcastableScalarConstant()493 *tensor = params->converter->CreateConstantLayer(weights, broadcastable_dims); in CreateBroadcastableScalarConstant()