Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes.cc397 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() local
427 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()