Searched refs:shapes2 (Results 1 – 2 of 2) sorted by relevance
1922 std::vector<TensorShape> shapes2{TensorShape({1})}; in TEST_F() local1925 Status status = compiler.SetDeviceToHostMetadata(key, types2, shapes2); in TEST_F()1951 std::vector<TensorShape> shapes2{TensorShape({1})}; in TEST_F() local1954 Status status = compiler.SetHostToDeviceMetadata(key, types2, shapes2); in TEST_F()
660 auto shapes2 = config2->mutable_program_shape(); in TEST() local661 *shapes2->add_parameters() = in TEST()663 *shapes2->add_parameters() = in TEST()665 *shapes2->mutable_result() = in TEST()