Home
last modified time | relevance | path

Searched defs:mask_shape (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/contrib/factorization/python/kernel_tests/
Dmasked_matmul_benchmark.py40 def _make_sparse_mask(self, mask_shape, nnz, sort=False): argument
/external/tensorflow/tensorflow/compiler/xla/service/
Ddynamic_padder.cc153 const Shape mask_shape = in Run() local
Dconvolution_group_converter.cc170 Shape mask_shape = ShapeUtil::MakeShape( in GetExpandedFilterMask() local