local_client_execute_test.cc | 137 auto y_array = LiteralToShapedBuffer(LiteralUtil::CreateR2WithLayout( local 139 EXPECT_TRUE(LayoutUtil::Equal(y_array.on_device_shape().layout(), 143 ExecuteLocallyOrDie(computation, {&x_array, &y_array}); 150 ExecuteLocallyOrDie(computation, {&y_array, &x_array}); 165 auto y_array = LiteralToShapedBuffer( local 170 computation, {&x_array, &y_array}, 182 computation, {&x_array, &y_array}, 202 auto y_array = LiteralToShapedBuffer( local 206 ExecuteLocallyOrDie(computation, {&x_array, &y_array}); 230 auto y_array = LiteralToShapedBuffer local [all...] |