Searched defs:diag_shape (Results 1 – 3 of 3) sorted by relevance
30 const TensorShape diag_shape = context->InputShape(1); in Compile() local
52 const TensorShape& diag_shape = diag.shape(); in Compute() local
63 def _gen_positive_diag(self, dtype, diag_shape): argument