Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dcollection_ops_util.cc301 auto per_slice_shape = llvm::to_vector<8>(buffer_type.getShape()); in ScatterAccumulateElements() local
302 per_slice_shape[0] = 1; in ScatterAccumulateElements()
303 auto slice_sizes = GetR1Const(per_slice_shape, builder, loc); in ScatterAccumulateElements()