Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes.cc436 int output_l[max_nb_dims], output_r[max_nb_dims]; in GetTrtBroadcastShape() local
438 output_l, operand_l_new_dims)); in GetTrtBroadcastShape()
445 if (!use_implicit_batch && (output_l[i] == -1 || output_r[i] == -1)) { in GetTrtBroadcastShape()
457 if ((output_l[i] != output_r[i]) && (output_l[i] != 1) && in GetTrtBroadcastShape()
460 "batch_dim: ", output_l[0], ", ", in GetTrtBroadcastShape()