Home
last modified time | relevance | path

Searched refs:broadcastable_dims (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes.cc475 nvinfer1::Dims broadcastable_dims(dims); in CreateBroadcastableScalarConstant() local
476 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()