Searched defs:vector_shape (Results 1 – 8 of 8) 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).value(); in XLA_TEST_F() local
159 ShapeHandle vector_shape = c->Vector(last_dim); in __anon84cbdd650302() local
1220 ShapeHandle vector_shape = c->Vector(channel_dim); in FusedBatchNormShape() local
5036 std::vector<int> vector_shape(dims_count); in PrepareOpsToDelegate() local
1361 Shape vector_shape = ShapeUtil::MakeShape(F32, {17}); in TEST_F() local
5287 std::vector<int> vector_shape(dims_count); in DensifyAndDequantizeConstTensor() local