Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes.cc396 auto compute_output_dims = [](const TRT_TensorOrWeights& input, in GetTrtBroadcastShape() local
427 TF_RETURN_IF_ERROR(compute_output_dims(operand_l, broadcast_num_dims, in GetTrtBroadcastShape()
429 TF_RETURN_IF_ERROR(compute_output_dims(operand_r, broadcast_num_dims, in GetTrtBroadcastShape()