Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes.cc4212 std::vector<int> bias_shape_vec(bias_shape.d, in ConvertBiasAdd() local
4215 bias_shape_vec.insert(bias_shape_vec.begin(), 1); in ConvertBiasAdd()
4217 bias_shape_vec.insert(bias_shape_vec.end(), in ConvertBiasAdd()
4218 input_shape.nbDims - bias_shape_vec.size(), 1); in ConvertBiasAdd()
4220 TensorShapeArrayToTrtDims(bias_shape_vec, &bias_shape)); in ConvertBiasAdd()