Searched refs:get_temp_weights (Results 1 - 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/contrib/tensorrt/convert/
H A Dconvert_nodes.cc358 TRT_ShapedWeights get_temp_weights(tensorflow::DataType type, function in class:tensorflow::tensorrt::convert::__anon26121::Converter
368 return this->get_temp_weights(weights.type_, weights.shape_);
679 TRT_ShapedWeights weights_output = ctx.get_temp_weights(dtype, output_shape);
1113 weights = ctx.get_temp_weights(dtype, GetTensorShape(tensor));

Completed in 177 milliseconds