Searched refs:lhs_arg_shape (Results 1 – 1 of 1) sorted by relevance
133 const Shape& lhs_arg_shape, in ExecuteAndFetchProfile() argument147 lhs_arg_shape, allocator, backend->default_device_ordinal())); in ExecuteAndFetchProfile()149 stream_ptr.get(), Literal::CreateFromShape(lhs_arg_shape), lhs_arg)); in ExecuteAndFetchProfile()162 client->Compile(computation, {&lhs_arg_shape, &rhs_arg_shape}, in ExecuteAndFetchProfile()