Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Ddot_handler.cc99 bool windowed_at_contracting_dims; member
140 int64 rhs_concat_dim, bool windowed_at_contracting_dims, in GenNewWindow() argument
150 if (windowed_at_contracting_dims) { in GenNewWindow()
168 !windowed_at_contracting_dims && !windowed_at_batch_dims && in GenNewWindow()
180 if (windowed_at_contracting_dims) { in GenNewWindow()
224 bool windowed_at_contracting_dims, bool windowed_at_batch_dims, in GenNewConvDNums() argument
251 if (rhs_concat_dim != -1 && !windowed_at_contracting_dims && in GenNewConvDNums()
276 if (lhs_concat_dim != -1 && !windowed_at_contracting_dims && in GenNewConvDNums()
286 if (!windowed_at_contracting_dims) { in GenNewConvDNums()
596 !einsum_config.windowed_at_contracting_dims); in PartitionBaseCase()
[all …]