Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes.cc5598 const auto check_weight_is_not_batched = in ConvertBatchMatMul() local
5611 TF_RETURN_IF_ERROR(check_weight_is_not_batched(inputs.at(0), inputs.at(1))); in ConvertBatchMatMul()
5612 TF_RETURN_IF_ERROR(check_weight_is_not_batched(inputs.at(1), inputs.at(0))); in ConvertBatchMatMul()