Searched refs:CheckInputsWeights (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes.cc | 2050 Status CheckInputsWeights( in CheckInputsWeights() function
2084 Status CheckInputsWeights( in CheckInputsWeights() function
2095 return CheckInputsWeights(params, expected_inputs); in CheckInputsWeights()
2280 TF_RETURN_IF_ERROR(CheckInputsWeights( in ConvertConv2DHelper()
2287 CheckInputsWeights(*params, {{"input", false}, {"filter", true}})); in ConvertConv2DHelper()
2493 CheckInputsWeights(*params, {{"x", false}, {"perm", true}})); in ConvertTranspose()
2535 CheckInputsWeights(*params, {{"input", TrtInputArg::kBoth}})); in ConvertShape()
2570 CheckInputsWeights(*params, {{"tensor", false}, {"shape", true}})); in ConvertReshape()
2641 CheckInputsWeights(*params, {{"input", false}, {"axis", true}})); in ConvertExpandDims()
2789 TF_RETURN_IF_ERROR(CheckInputsWeights(*params, {{"input", false}})); in ConvertSqueeze()
[all …]
|