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.cc395 auto compute_output_dims = [use_implicit_batch]( in GetTrtBroadcastShape() local
437 TF_RETURN_IF_ERROR(compute_output_dims(operand_l, broadcast_num_dims, in GetTrtBroadcastShape()
439 TF_RETURN_IF_ERROR(compute_output_dims(operand_r, broadcast_num_dims, in GetTrtBroadcastShape()