Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes.cc2341 const bool need_transpose = (data_format == "NHWC"); in ConvertConv2DHelper() local
2342 if (need_transpose) { in ConvertConv2DHelper()
2467 if (need_transpose) { in ConvertConv2DHelper()
2982 const bool need_transpose = !(pad_dims[0] == 2 && pad_dims[1] == 3); in ConvertStridedSliceHelper() local
2985 if (need_transpose) { in ConvertStridedSliceHelper()
3003 if (need_transpose) { in ConvertStridedSliceHelper()
3016 if (need_transpose) { in ConvertStridedSliceHelper()
3335 const bool need_transpose = is_ndhwc; in ConvertConv3DHelper() local
3336 if (need_transpose) { in ConvertConv3DHelper()
3400 if (need_transpose) { in ConvertConv3DHelper()
[all …]