Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Ddeep_conv2d.cc235 const int64 shard_base = shard_stride * (s_r * shard_cols + s_c); in operator ()() local
239 i * in_stride + out_depth_buf_base + shard_base; in operator ()()
240 const int64 out_base = i * out_stride + out_depth_base + shard_base; in operator ()()
330 const int64 shard_base = shard_stride * (s_r * shard_cols + s_c); in operator ()() local
347 out_depth_base + shard_base; in operator ()()
787 const int64 shard_base = sr * filter_shards_col + sc; in operator ()() local
788 const int64 out_buf_base = tile_base + out_depth_base + shard_base; in operator ()()