Searched refs:broadcast_num_dims (Results 1 – 1 of 1) sorted by relevance
397 int broadcast_num_dims, int* output_dims_array, in GetTrtBroadcastShape()402 output_dims_array + broadcast_num_dims - input_dims.nbDims); in GetTrtBroadcastShape()405 if (true_input_dims < broadcast_num_dims) { in GetTrtBroadcastShape()409 broadcast_num_dims, ")"); in GetTrtBroadcastShape()418 output_dims->nbDims = broadcast_num_dims - 1; in GetTrtBroadcastShape()419 std::copy(output_dims_array + 1, output_dims_array + broadcast_num_dims, in GetTrtBroadcastShape()422 output_dims->nbDims = broadcast_num_dims; in GetTrtBroadcastShape()423 std::copy(output_dims_array, output_dims_array + broadcast_num_dims, in GetTrtBroadcastShape()431 const int broadcast_num_dims = in GetTrtBroadcastShape() local437 TF_RETURN_IF_ERROR(compute_output_dims(operand_l, broadcast_num_dims, in GetTrtBroadcastShape()[all …]