Searched refs:device_name_t (Results 1 – 2 of 2) sorted by relevance
39 Tensor* device_name_t; in Compute() local41 ctx->allocate_output(0, TensorShape({}), &device_name_t)); in Compute()45 device_name_t->scalar<tstring>()() = ctx->device()->name(); in Compute()
905 Tensor* device_name_t; in Compute() local907 ctx->allocate_output(0, TensorShape({}), &device_name_t)); in Compute()917 device_name_t->scalar<int32>()() = index; in Compute()