Searched refs:broadcasted_dims_r (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()
4487 params->converter, operand_r, broadcasted_dims_r, 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()
5630 PrepareTensorForShape(params->converter, inputs.at(1), broadcasted_dims_r, 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()
5888 PrepareTensorForShape(params->converter, inputs.at(1), broadcasted_dims_r, in ConvertSquaredDifference()
|