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