Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes.h280 bool is_weights() const { return initialized_ && !is_tensor_; } in is_weights() function
287 CHECK(is_weights()); in weights()
292 CHECK(is_weights()); in weights()
Dconvert_nodes.cc1446 if (!inputs_is_weight[i].second && inputs.at(i).is_weights()) { in CheckInputsWeights()
3158 if (inputs.at(0).is_weights() && inputs.at(1).is_weights()) { in ConvertBinary()
3177 if (inputs.at(0).is_tensor() && inputs.at(1).is_weights()) { in ConvertBinary()
3180 } else if (inputs.at(0).is_weights() && inputs.at(1).is_tensor()) { in ConvertBinary()
3866 if (inputs[0].is_weights() && inputs[1].is_weights()) { in ConvertBatchMatMul()
3876 inputs.at(1).is_weights()) { in ConvertBatchMatMul()
3889 if (input.is_weights()) { in ConvertBatchMatMul()
Dconvert_nodes_test.cc329 EXPECT_EQ(false, ptr->is_weights()); in TEST()
350 EXPECT_EQ(false, ptr->is_weights()); in TEST()
373 EXPECT_EQ(false, ptr->is_weights()); in TEST()
388 EXPECT_EQ(true, ptr->is_weights()); in TEST()