Searched refs:shape_arg (Results 1 – 1 of 1) sorted by relevance
858 const Shape& shape_arg = replicated_arguments.front()[i]->on_host_shape(); in Execute() local859 if (!ShapeUtil::Equal(shape_module, shape_arg)) { in Execute()863 ShapeUtil::HumanStringWithLayout(shape_arg)); in Execute()