Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes.cc2275 TRT_TensorOrWeights backprop_output_size; in ConvertConv2DHelper() local
2283 backprop_output_size = inputs.at(0); in ConvertConv2DHelper()
2377 GetSpatialDimsFromOutputSizes(backprop_output_size, h_index, w_index); in ConvertConv2DHelper()
2437 GetSpatialDimsFromOutputSizes(backprop_output_size, h_index, w_index); in ConvertConv2DHelper()
3230 TRT_TensorOrWeights backprop_output_size; in ConvertConv3DHelper() local
3238 backprop_output_size = inputs.at(0); in ConvertConv3DHelper()
3313 static_cast<int*>(backprop_output_size.weights().GetValues()); in ConvertConv3DHelper()