Searched refs:stride_config (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | literal.h | 1090 StrideConfig stride_config(this_shape, this_shape, in PopulateInternal()
1093 ShapeUtil::GetDimension(this_shape, stride_config.minor_dimension); in PopulateInternal()
1101 minor_scan_indexes[stride_config.minor_dimension] = i; in PopulateInternal()
1106 ShapeUtil::ForEachIndexParallel(this_shape, stride_config.base, in PopulateInternal()
1107 stride_config.dimensions, in PopulateInternal()
1108 stride_config.step, init_function); in PopulateInternal()
1111 this_shape, stride_config.base, stride_config.dimensions, in PopulateInternal()
1112 stride_config.step, in PopulateInternal()
|
| D | literal.cc | 240 MutableLiteralBase::StrideConfig stride_config(src_literal.shape(), shape(), in CopySliceFromInternal() local
255 StridedCopy(this->data<NativeT>(), dest_index, stride_config.dest_stride, in CopySliceFromInternal()
257 stride_config.source_stride, stride_config.minor_loop_size); in CopySliceFromInternal()
261 ShapeUtil::ForEachIndex(src_literal.shape(), stride_config.base, in CopySliceFromInternal()
262 stride_config.dimensions, stride_config.step, in CopySliceFromInternal()
|