Searched defs:diag_shape (Results 1 – 5 of 5) sorted by relevance
244 const TensorShape diag_shape = context->InputShape(0); in Compile() local467 const TensorShape diag_shape = context->InputShape(1); in Compile() local
89 const TensorShape& diag_shape = diag.shape(); in Compute() local
65 def _gen_positive_diag(self, dtype, diag_shape): argument
719 def _testGrad(self, input_shape, diag_shape, diags, align): argument
1423 ShapeHandle input_shape, diag_shape, diag_index_shape; in MatrixSetDiagV2Shape() local