Searched defs:token_shape (Results 1 – 5 of 5) sorted by relevance
254 auto token_shape = ShapeUtil::MakeTokenShape(); in TEST_F() local303 auto token_shape = ShapeUtil::MakeTokenShape(); in TEST_F() local
56 std::vector<executorch::aten::SizesType> token_shape; in generate() local
257 std::vector<SizesType> token_shape = {1, 1}; in generate() local
1170 xla::Shape token_shape = xla::TypeToShape(result_types[num_results - 1]); in ExportXlaOp() local1283 xla::Shape token_shape = xla::TypeToShape(result_types[num_results - 1]); in ExportXlaOp() local
429 auto token_shape = ShapeUtil::MakeTokenShape(); in XLA_TEST_F() local