Searched refs:expected_diag_shape (Results 1 – 2 of 2) sorted by relevance
134 TensorShape expected_diag_shape = input_shape; in Compute() local135 expected_diag_shape.RemoveLastDims(2); in Compute()136 if (num_diags > 1) expected_diag_shape.AddDim(num_diags); in Compute()140 expected_diag_shape.AddDim(max_diag_len); in Compute()142 context, expected_diag_shape == diag_shape, in Compute()149 "\nExpected diagonal shape: ", expected_diag_shape.DebugString())); in Compute()
504 TensorShape expected_diag_shape = input_shape; in Compile() local505 expected_diag_shape.RemoveLastDims(2); in Compile()506 if (num_diags > 1) expected_diag_shape.AddDim(num_diags); in Compile()510 expected_diag_shape.AddDim(max_diag_len); in Compile()512 context, expected_diag_shape == diag_shape, in Compile()519 "\nExpected diagonal shape: ", expected_diag_shape.DebugString())); in Compile()