Searched defs:shape_arg (Results 1 – 2 of 2) sorted by relevance
921 const Shape& shape_arg = replicated_arguments.front()[i]->on_device_shape(); in Execute() local
446 auto shape_arg = shape_arguments->at(*norm_index); in substituteSymbolicProperties() local