Searched refs:broadcasted_dims_l (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes.cc | 4476 nvinfer1::Dims broadcasted_dims_l, broadcasted_dims_r; in ConvertBinary() local
4479 params->use_implicit_batch, &broadcasted_dims_l, &broadcasted_dims_r)); in ConvertBinary()
4484 params->converter, operand_l, broadcasted_dims_l, params->validation_only, in ConvertBinary()
5620 nvinfer1::Dims broadcasted_dims_l, broadcasted_dims_r; in ConvertBatchMatMul() local
5623 params->use_implicit_batch, &broadcasted_dims_l, &broadcasted_dims_r)); in ConvertBatchMatMul()
5627 PrepareTensorForShape(params->converter, inputs.at(0), broadcasted_dims_l, in ConvertBatchMatMul()
5878 nvinfer1::Dims broadcasted_dims_l, broadcasted_dims_r; in ConvertSquaredDifference() local
5881 params->use_implicit_batch, &broadcasted_dims_l, &broadcasted_dims_r)); in ConvertSquaredDifference()
5885 PrepareTensorForShape(params->converter, inputs.at(0), broadcasted_dims_l, in ConvertSquaredDifference()
|