Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ops/
H A Deinsum.cc408 const EinsumDescriptor& desc, const bool need_transpose) { in ConditionEinsumWeights()
432 const bool need_transpose, in ConditionEinsumTensor()
468 bool need_transpose = !desc.permute.empty(); in ConditionEinsumOperand() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/
H A Dconv_weights_converter.cc133 const bool need_transpose = in GetConverterToConvWeightsCode() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
H A Dconvert_nodes.cc1895 const bool need_transpose = (data_format == "NHWC"); in ConvertConv2DHelper() local
2905 const bool need_transpose = is_ndhwc; in ConvertConv3DHelper() local
3154 const bool need_transpose = (data_format == "NHWC"); in ConvertFusedConv2DBiasActivation() local