Searched refs:GetDeviceAliasForLogicalCore (Results 1 – 5 of 5) sorted by relevance
98 if (!devices.getValue().get(tensorflow::GetDeviceAliasForLogicalCore(0))) in ColocateCompositeResourceOpsInReplicate()105 tensorflow::GetDeviceAliasForLogicalCore(0)); in ColocateCompositeResourceOpsInReplicate()
419 builder->getNamedAttr(tensorflow::GetDeviceAliasForLogicalCore(core), in AssignDevicesToReplicate()573 ? tensorflow::GetDeviceAliasForLogicalCore(core) in AddToParallelExecuteOp()593 std::string device = replicated ? tensorflow::GetDeviceAliasForLogicalCore(0) in AssignDevicesToReplicatedExecute()
235 assert(devices.find(tensorflow::GetDeviceAliasForLogicalCore(0)) in AddInputsToReplicateOp()296 devices.find(tensorflow::GetDeviceAliasForLogicalCore(0))->getSecond(); in CreateStateVars()
243 std::string GetDeviceAliasForLogicalCore(int core_index);
481 std::string GetDeviceAliasForLogicalCore(int core_index) { in GetDeviceAliasForLogicalCore() function