Searched refs:compute_output_dims (Results 1 – 1 of 1) sorted by relevance
396 auto compute_output_dims = [](const TRT_TensorOrWeights& input, in GetTrtBroadcastShape() local427 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()