Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes.h509 nvinfer1::Dims* operand_l_new_dims,
Dconvert_nodes.cc364 nvinfer1::Dims* operand_l_new_dims, in GetTrtBroadcastShape() argument
428 output_l, operand_l_new_dims)); in GetTrtBroadcastShape()
438 DebugString(*operand_l_new_dims), " vs ", "batch_dim: ", output_r[0], in GetTrtBroadcastShape()