Searched refs:num_computation_num_args_buffers (Results 1 – 1 of 1) sorted by relevance
482 std::vector<std::vector<xla::PjRtBuffer*>> num_computation_num_args_buffers( in Call() local485 num_computation_num_args_buffers[computation].resize(num_args); in Call()496 num_computation_num_args_buffers[computation][i] = in Call()513 num_computation_num_args_buffers, in Call()