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

/external/tensorflow/tensorflow/contrib/tensorrt/convert/
H A Dconvert_nodes.cc178 bool is_weights() const { return variant_ == TRT_NODE_WEIGHTS; } function in class:tensorflow::tensorrt::convert::__anon26121::TRT_TensorOrWeights
189 CHECK_EQ(is_weights(), true);
193 CHECK_EQ(is_weights(), true);
1037 !inputs.at(1).is_weights())
1147 if (inputs.at(0).is_weights() && inputs.at(1).is_weights())
1150 if (inputs.at(0).is_tensor() && inputs.at(1).is_weights())
1154 if (inputs.at(0).is_weights() && inputs.at(1).is_tensor())
1174 if (inputs.at(0).is_weights())
1189 !inputs.at(1).is_weights())
[all...]

Completed in 66 milliseconds