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()
|