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