Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/
Dindex_util.h69 static int64 GetDimensionStride(const Shape& shape, int64 dimension);
Dindex_util.cc129 /* static */ int64 IndexUtil::GetDimensionStride(const Shape& shape, in GetDimensionStride() function in xla::IndexUtil
Dliteral.cc96 dest_stride = IndexUtil::GetDimensionStride(dest_shape, minor_dimension); in StrideConfig()
100 IndexUtil::GetDimensionStride(source_shape, minor_dimension); in StrideConfig()