Searched refs:host_ordinal_to_global_device_id_tensor (Results 1 – 1 of 1) sorted by relevance
175 const Tensor& host_ordinal_to_global_device_id_tensor = ctx->input(i); in Compute() local177 ctx, host_ordinal_to_global_device_id_tensor.dims() == 1, in Compute()179 host_ordinal_to_global_device_id_tensor.dims(), in Compute()189 const Tensor& host_ordinal_to_global_device_id_tensor = ctx->input(i); in Compute() local191 host_ordinal_to_global_device_id_tensor.flat<int>(); in Compute()194 host_ordinal_to_global_device_id_tensor.dim_size(0); in Compute()198 host_ordinal_to_global_device_id_tensor.dim_size(0), in Compute()201 host_ordinal_to_global_device_id_tensor.dim_size(0), in Compute()204 for (int j = 0; j < host_ordinal_to_global_device_id_tensor.dim_size(0); in Compute()