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