Searched defs:token_shape (Results 1 – 3 of 3) sorted by relevance
254 auto token_shape = ShapeUtil::MakeTokenShape(); in TEST_F() local303 auto token_shape = ShapeUtil::MakeTokenShape(); in TEST_F() local
429 auto token_shape = ShapeUtil::MakeTokenShape(); in XLA_TEST_F() 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