Searched defs:shape_in (Results 1 – 4 of 4) sorted by relevance
44 TensorShape shape_in = logits_in.shape(); in Compute() local
866 Shape shape_in = ShapeUtil::MakeShape(F32, {16, 8}); in TEST_F() local905 Shape shape_in = ShapeUtil::MakeShape(C64, {16, 5}); in TEST_F() local
475 ShapeHandle shape_in = c->input(1); in __anondb9326b20702() local
619 auto shape_in = properties.GetOutputProperties("ones").at(0).shape(); in TEST_F() local