Searched defs:vector_shape (Results 1 – 6 of 6) sorted by relevance
37 Shape vector_shape = ShapeUtil::MakeShape(F32, {100}); in TEST() local
158 ShapeHandle vector_shape = c->Vector(last_dim); in __anon3e672dd80402() local2412 ShapeHandle vector_shape = c->Vector(channel_dim); in __anon3e672dd83e02() local
280 const Shape vector_shape = ShapeUtil::MakeShape(F32, {3}); in XLA_TEST_F() local326 const Shape vector_shape = ShapeUtil::MakeShape(F32, {3}); in XLA_TEST_F() local
170 auto vector_shape = builder.GetShape(vector_element).ConsumeValueOrDie(); in XLA_TEST_F() local
816 ShapeHandle vector_shape = c->Vector(channel_dim); in FusedBatchNormShape() local
879 Shape vector_shape = ShapeUtil::MakeShape(F32, {17}); in TEST_F() local