Searched defs:logits_shape (Results 1 – 6 of 6) sorted by relevance
40 const TensorShape logits_shape = ctx->InputShape(0); in Compile() local148 const TensorShape logits_shape = ctx->InputShape(0); in Compile() local181 const TensorShape logits_shape = ctx->InputShape(0); in Compile() local
43 TensorShape logits_shape = ctx->InputShape(0); in Compile() local
86 ShapeHandle logits_shape; in __anon530aab340202() local
114 ShapeHandle logits_shape; in __anon979c5d7d0302() local
206 auto logits_shape = in __anone3ce78ae0702() local277 auto logits_shape = in __anone3ce78ae0a02() local
116 TensorShape logits_shape({5, 3}); in TEST_F() local