Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes.cc4990 const int split_size_on_axis = dims.d[trt_axis] / num_splits; in ConvertSplitHelper() local
4991 size[trt_axis] = split_size_on_axis; in ConvertSplitHelper()
5012 begin[trt_axis + 1] = i * split_size_on_axis; in ConvertSplitHelper()