Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
Doptimize.cc737 auto channel_index_iter = filter_shape.begin(); in GetBiasDimAndSliceSize() local
738 std::advance(channel_index_iter, op.GetChannelDimIndex()); in GetBiasDimAndSliceSize()
741 std::accumulate(std::next(channel_index_iter), filter_shape.end(), 1, in GetBiasDimAndSliceSize()
743 return {*channel_index_iter, slice_size}; in GetBiasDimAndSliceSize()