Searched refs:expected_out_shape (Results 1 – 2 of 2) sorted by relevance
341 TensorShape expected_out_shape; in Compile() local348 &expected_out_shape)); in Compile()349 OP_REQUIRES(ctx, expected_out_shape == out_backprop_shape, in Compile()
1942 expected_out_shape = indices_shape[:-1]1944 expected_out_shape += params_shape[-ndims_rows:]1945 self.assertSequenceEqual(out.shape, expected_out_shape)2103 expected_out_shape = indices_shape[:-1]2105 expected_out_shape += params_shape[-row_ndims:]2107 self.assertSequenceEqual(res.shape, expected_out_shape)