Searched refs:device_id_output (Results 1 – 1 of 1) sorted by relevance
292 int32_t* device_id_output = nullptr; in Compute() local293 auto cleanup = xla::MakeCleanup([&status, &device_id_output]() { in Compute()295 tpu::OpsApiFn()->TpuConfigurationApi_FreeInt32ArrayFn(device_id_output); in Compute()306 params.core_id_output = &device_id_output; in Compute()364 ctx_output->flat<int32>()(i) = device_id_output[i]; in Compute()