Searched defs:diag_shape (Results 1 – 5 of 5) sorted by relevance
247 const TensorShape diag_shape = context->InputShape(0); in Compile() local470 const TensorShape diag_shape = context->InputShape(1); in Compile() local
92 const TensorShape& diag_shape = diag.shape(); in Compute() local
62 def _gen_positive_diag(self, dtype, diag_shape): argument
715 def _testGrad(self, input_shape, diag_shape, diags, align): argument
1514 ShapeHandle input_shape, diag_shape, diag_index_shape; in MatrixSetDiagV2Shape() local