Searched refs:on_device_executable_parameter_shapes_ (Results 1 – 2 of 2) sorted by relevance
846 std::vector<std::vector<Shape>> on_device_executable_parameter_shapes_; variable
1787 on_device_executable_parameter_shapes_.push_back( in PjRtStreamExecutorExecutable()1972 on_device_executable_parameter_shapes_[executable_idx], in EnqueueExecution()