Searched refs:host_ordinal_to_global_device_id_tensor (Results 1 – 1 of 1) sorted by relevance
153 const Tensor& host_ordinal_to_global_device_id_tensor = ctx->input(i); in Compute() local155 ctx, host_ordinal_to_global_device_id_tensor.dims() == 1, in Compute()157 host_ordinal_to_global_device_id_tensor.dims(), in Compute()167 const Tensor& host_ordinal_to_global_device_id_tensor = ctx->input(i); in Compute() local169 host_ordinal_to_global_device_id_tensor.flat<int>(); in Compute()172 host_ordinal_to_global_device_id_tensor.dim_size(0); in Compute()176 host_ordinal_to_global_device_id_tensor.dim_size(0), in Compute()179 host_ordinal_to_global_device_id_tensor.dim_size(0), in Compute()182 for (int j = 0; j < host_ordinal_to_global_device_id_tensor.dim_size(0); in Compute()