Searched refs:CheckInputsWeights (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes.cc | 1426 Status CheckInputsWeights( in CheckInputsWeights() function
1772 TF_RETURN_IF_ERROR(CheckInputsWeights( in ConvertConv2DHelper()
1779 CheckInputsWeights(*params, {{"input", false}, {"filter", true}})); in ConvertConv2DHelper()
2046 CheckInputsWeights(*params, {{"x", false}, {"perm", true}})); in ConvertTranspose()
2082 CheckInputsWeights(*params, {{"tensor", false}, {"shape", true}})); in ConvertReshape()
2181 CheckInputsWeights(*params, {{"input", false}, {"axis", true}})); in ConvertExpandDims()
2233 TF_RETURN_IF_ERROR(CheckInputsWeights(*params, {{"input", false}})); in ConvertSqueeze()
2498 TF_RETURN_IF_ERROR(CheckInputsWeights( in ConvertSlice()
2543 TF_RETURN_IF_ERROR(CheckInputsWeights( in ConvertStridedSlice()
2642 TF_RETURN_IF_ERROR(CheckInputsWeights(*params, {{"input", false}})); in ConvertPool()
[all …]
|