Searched refs:Convert2DCBuffersToCppBuffers (Results 1 – 2 of 2) sorted by relevance
384 static std::vector<std::vector<xla::PjRtBuffer*>> Convert2DCBuffersToCppBuffers( in Convert2DCBuffersToCppBuffers() function413 Convert2DCBuffersToCppBuffers(args->argument_lists, args->num_devices, in PJRT_Executable_Execute()
475 Convert2DCBuffersToCppBuffers(PJRT_Buffer*** c_lists, size_t outer_size, in Convert2DCBuffersToCppBuffers() function537 return Convert2DCBuffersToCppBuffers(args.output_lists, args.num_devices, in Execute()