Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dmlir_emitter.cc76 int64_t accumulated_stride = 1; in BuildViewForBuffer() local
79 strides[dim] = accumulated_stride; in BuildViewForBuffer()
80 accumulated_stride *= opShape.dimensions(dim); in BuildViewForBuffer()
/external/tensorflow/tensorflow/compiler/mlir/xla/
Dhlo_utils.cc57 int64_t accumulated_stride = 1; in GetPermutationIfAvailable() local
60 strides[dim] = accumulated_stride; in GetPermutationIfAvailable()
61 accumulated_stride *= shape.dimensions(dim); in GetPermutationIfAvailable()
63 if (accumulated_stride == 0) { in GetPermutationIfAvailable()