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()416 output_dims->nbDims = broadcast_num_dims - 1; in GetTrtBroadcastShape()417 std::copy(output_dims_array + 1, output_dims_array + broadcast_num_dims, in GetTrtBroadcastShape()423 const int broadcast_num_dims = in GetTrtBroadcastShape() local427 TF_RETURN_IF_ERROR(compute_output_dims(operand_l, broadcast_num_dims, in GetTrtBroadcastShape()429 TF_RETURN_IF_ERROR(compute_output_dims(operand_r, broadcast_num_dims, in GetTrtBroadcastShape()433 for (int i = 0; i < broadcast_num_dims; ++i) { in GetTrtBroadcastShape()