Searched defs:batch_shape_ (Results 1 – 1 of 1) sorted by relevance
42 def _GetMatrixBandPartTest(dtype_, batch_shape_, shape_): argument72 def _GetMatrixBandPartGradTest(dtype_, batch_shape_, shape_): argument