Searched defs:vector_shape (Results 1 – 7 of 7) sorted by relevance
37 Shape vector_shape = ShapeUtil::MakeShape(F32, {100}); in TEST() local
283 const Shape vector_shape = ShapeUtil::MakeShape(F32, {3}); in XLA_TEST_F() local329 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
155 ShapeHandle vector_shape = c->Vector(last_dim); in __anonad10f3360302() local
3174 std::vector<int> vector_shape(dims_count); in PrepareOpsToDelegate() local
1165 ShapeHandle vector_shape = c->Vector(channel_dim); in FusedBatchNormShape() local
1326 Shape vector_shape = ShapeUtil::MakeShape(F32, {17}); in TEST_F() local